python 批量保存到mysql

原创

mob649e8155edc4 2023-12-17 11:28:56 ©著作权

文章标签 MySQL 数据库 python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e8155edc4的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 批量保存到MySQL

MySQL是一种常用的关系型数据库管理系统，而Python是一种简单易用的编程语言。在很多情况下，我们需要将Python中的数据批量保存到MySQL数据库中。本文将介绍如何使用Python来实现这一功能，并提供相应的代码示例。

准备工作

在使用Python连接MySQL之前，我们需要确保已经安装了相应的模块。可以通过以下命令来安装：

pip install mysql-connector-python

安装完成后，我们可以使用import mysql.connector语句来导入相应的模块。

连接数据库

在将数据保存到MySQL之前，我们首先需要连接到数据库。使用以下代码来建立与MySQL数据库的连接：

import mysql.connector

# 建立数据库连接
cnx = mysql.connector.connect(user='<username>', password='<password>',
                              host='<hostname>',
                              database='<database>')

# 获取数据库游标
cursor = cnx.cursor()

以上代码中，<username>、<password>、<hostname>和<database>分别代表用户名、密码、主机名和数据库名。根据实际情况进行修改。

创建表格

在将数据保存到数据库之前，我们需要创建一个表格来存储这些数据。可以使用以下代码来创建一个名为<table>的表格：

# 创建表格
create_table_query = '''
    CREATE TABLE <table> (
        id INT PRIMARY KEY AUTO_INCREMENT,
        name VARCHAR(255),
        age INT
    )
'''

cursor.execute(create_table_query)

以上代码中，<table>代表表格名，根据实际情况进行修改。在这个例子中，我们创建了一个包含id、name和age字段的表格。

保存数据

一旦表格创建成功，我们就可以将数据保存到表格中。假设我们有一个包含多个字典的列表，每个字典代表一个数据条目。可以使用以下代码将数据保存到表格中：

# 数据列表
data = [
    {'name': 'Alice', 'age': 25},
    {'name': 'Bob', 'age': 30},
    {'name': 'Charlie', 'age': 35}
]

# 插入数据
insert_query = '''
    INSERT INTO <table> (name, age)
    VALUES (%s, %s)
'''

for item in data:
    cursor.execute(insert_query, (item['name'], item['age']))

以上代码中，<table>代表表格名，根据实际情况进行修改。在这个例子中，我们将name和age字段插入到表格中。

提交更改

在将数据保存到表格中后，我们需要提交更改以确保数据被持久化到数据库中。可以使用以下代码来提交更改：

cnx.commit()

关闭连接

最后，我们需要关闭与数据库的连接。可以使用以下代码来关闭连接：

cursor.close()
cnx.close()

完整示例代码

import mysql.connector

# 建立数据库连接
cnx = mysql.connector.connect(user='<username>', password='<password>',
                              host='<hostname>',
                              database='<database>')

# 获取数据库游标
cursor = cnx.cursor()

# 创建表格
create_table_query = '''
    CREATE TABLE <table> (
        id INT PRIMARY KEY AUTO_INCREMENT,
        name VARCHAR(255),
        age INT
    )
'''

cursor.execute(create_table_query)

# 数据列表
data = [
    {'name': 'Alice', 'age': 25},
    {'name': 'Bob', 'age': 30},
    {'name': 'Charlie', 'age': 35}
]

# 插入数据
insert_query = '''
    INSERT INTO <table> (name, age)
    VALUES (%s, %s)
'''

for item in data:
    cursor.execute(insert_query, (item['name'], item['age']))

# 提交更改
cnx.commit()

# 关闭连接
cursor.close()
cnx.close()

以上就是使用Python批量保存数据到MySQL数据库的完整示例。在实际使用中，只需要根据实际情况修改相应的参数即可。希望本文对于理解如何使用Python保存数据到MySQL有所帮助。