Python写入MySQL数据库的方法有：使用MySQL Connector、使用SQLAlchemy、使用pymysql、使用MySQLdb。其中，使用MySQL Connector是最常见和推荐的方式，因为它是官方提供的库，性能和兼容性较好。下面将详细介绍如何使用MySQL Connector将数据写入MySQL数据库。
一、使用MySQL Connector写入MySQL数据库
MySQL Connector是由MySQL官方提供的Python库，用于连接和操作MySQL数据库。它具备良好的性能和兼容性，是推荐使用的方式。
1、安装MySQL Connector
在使用MySQL Connector之前，需要先安装该库。可以使用pip进行安装：
pip install mysql-connector-python
2、连接到MySQL数据库
在写入数据之前，首先需要连接到MySQL数据库。可以使用MySQL Connector提供的connect方法进行连接。以下是一个示例：
import mysql.connector
连接到MySQL数据库
conn = mysql.connector.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
创建一个游标对象
cursor = conn.cursor()
3、创建表
在写入数据之前，需要确保目标表已经存在。如果表不存在，可以使用SQL语句创建表。以下是一个创建表的示例：
# 创建表的SQL语句
create_table_query = """
CREATE TABLE IF NOT EXISTS employees (
id INT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(255) NOT NULL,
salary DECIMAL(10, 2),
department VARCHAR(255)
)
"""
执行创建表的SQL语句
cursor.execute(create_table_query)
4、插入数据
可以使用INSERT INTO SQL语句将数据插入到表中。以下是插入数据的示例：
# 插入数据的SQL语句
insert_query = """
INSERT INTO employees (name, salary, department)
VALUES (%s, %s, %s)
"""
要插入的数据
data = ("John Doe", 50000.00, "Engineering")
执行插入数据的SQL语句
cursor.execute(insert_query, data)
提交事务
conn.commit()
获取插入数据后的行ID
last_id = cursor.lastrowid
print(f"Inserted data with ID: {last_id}")
5、批量插入数据
如果需要插入大量数据，可以使用批量插入的方式来提高效率。以下是批量插入数据的示例：
# 批量插入数据的SQL语句
insert_query = """
INSERT INTO employees (name, salary, department)
VALUES (%s, %s, %s)
"""
要插入的数据列表
data = [
("Alice", 60000.00, "Marketing"),
("Bob", 45000.00, "Sales"),
("Charlie", 55000.00, "HR")
]
执行批量插入数据的SQL语句
cursor.executemany(insert_query, data)
提交事务
conn.commit()
6、关闭连接
在完成所有操作之后，需要关闭游标和数据库连接：
# 关闭游标
cursor.close()
关闭数据库连接
conn.close()
二、使用SQLAlchemy写入MySQL数据库
SQLAlchemy是一个功能强大的ORM（对象关系映射）库，它不仅支持MySQL，还支持其他多种数据库。使用SQLAlchemy可以更加简洁地进行数据库操作。
1、安装SQLAlchemy
可以使用pip安装SQLAlchemy：
pip install SQLAlchemy
pip install pymysql
2、连接到MySQL数据库
在写入数据之前，首先需要连接到MySQL数据库。可以使用SQLAlchemy提供的create_engine方法进行连接。以下是一个示例：
from sqlalchemy import create_engine
创建数据库连接引擎
engine = create_engine("mysql+pymysql://yourusername:yourpassword@localhost/yourdatabase")
创建数据库连接
conn = engine.connect()
3、定义表结构
在写入数据之前，需要定义目标表的结构。可以使用SQLAlchemy的Table和Column类来定义表结构。以下是一个定义表结构的示例：
from sqlalchemy import Table, Column, Integer, String, MetaData, DECIMAL
定义元数据对象
metadata = MetaData()
定义表结构
employees = Table(
'employees', metadata,
Column('id', Integer, primary_key=True, autoincrement=True),
Column('name', String(255), nullable=False),
Column('salary', DECIMAL(10, 2)),
Column('department', String(255))
)
创建表
metadata.create_all(engine)
4、插入数据
可以使用SQLAlchemy的insert方法将数据插入到表中。以下是插入数据的示例：
from sqlalchemy import insert
插入数据
insert_query = insert(employees).values(name="John Doe", salary=50000.00, department="Engineering")
执行插入数据的SQL语句
conn.execute(insert_query)
5、批量插入数据
如果需要插入大量数据，可以使用批量插入的方式来提高效率。以下是批量插入数据的示例：
# 要插入的数据列表
data = [
{"name": "Alice", "salary": 60000.00, "department": "Marketing"},
{"name": "Bob", "salary": 45000.00, "department": "Sales"},
{"name": "Charlie", "salary": 55000.00, "department": "HR"}
]
执行批量插入数据的SQL语句
conn.execute(employees.insert(), data)
6、关闭连接
在完成所有操作之后，需要关闭数据库连接：
# 关闭数据库连接
conn.close()
三、使用pymysql写入MySQL数据库
pymysql是一个纯Python实现的MySQL客户端库，支持Python 3.x版本。它的使用方法类似于MySQL Connector。
1、安装pymysql
可以使用pip安装pymysql：
pip install pymysql
2、连接到MySQL数据库
在写入数据之前，首先需要连接到MySQL数据库。可以使用pymysql提供的connect方法进行连接。以下是一个示例：
import pymysql
连接到MySQL数据库
conn = pymysql.connect(
host="localhost",
user="yourusername",
password="yourpassword",
database="yourdatabase"
)
创建一个游标对象
cursor = conn.cursor()
3、创建表
在写入数据之前，需要确保目标表已经存在。如果表不存在，可以使用SQL语句创建表。以下是一个创建表的示例：
# 创建表的SQL语句
create_table_query = """
CREATE TABLE IF NOT EXISTS employees (
id INT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(255) NOT NULL,
salary DECIMAL(10, 2),
department VARCHAR(255)
)
"""
执行创建表的SQL语句
cursor.execute(create_table_query)
4、插入数据
可以使用INSERT INTO SQL语句将数据插入到表中。以下是插入数据的示例：
# 插入数据的SQL语句
insert_query = """
INSERT INTO employees (name, salary, department)
VALUES (%s, %s, %s)
"""
要插入的数据
data = ("John Doe", 50000.00, "Engineering")
执行插入数据的SQL语句
cursor.execute(insert_query, data)
提交事务
conn.commit()
获取插入数据后的行ID
last_id = cursor.lastrowid
print(f"Inserted data with ID: {last_id}")
5、批量插入数据
如果需要插入大量数据，可以使用批量插入的方式来提高效率。以下是批量插入数据的示例：
# 批量插入数据的SQL语句
insert_query = """
INSERT INTO employees (name, salary, department)
VALUES (%s, %s, %s)
"""
要插入的数据列表
data = [
("Alice", 60000.00, "Marketing"),
("Bob", 45000.00, "Sales"),
("Charlie", 55000.00, "HR")
]
执行批量插入数据的SQL语句
cursor.executemany(insert_query, data)
提交事务
conn.commit()
6、关闭连接
在完成所有操作之后，需要关闭游标和数据库连接：
# 关闭游标
cursor.close()
关闭数据库连接
conn.close()
四、使用MySQLdb写入MySQL数据库
MySQLdb是一个用于Python 2.x版本的MySQL客户端库，不推荐在新的项目中使用，因为它不支持Python 3.x版本。以下是MySQLdb的使用方法，仅供参考。
1、安装MySQLdb
可以使用pip安装MySQLdb：
pip install MySQL-python
2、连接到MySQL数据库
在写入数据之前，首先需要连接到MySQL数据库。可以使用MySQLdb提供的connect方法进行连接。以下是一个示例：
import MySQLdb
连接到MySQL数据库
conn = MySQLdb.connect(
host="localhost",
user="yourusername",
passwd="yourpassword",
db="yourdatabase"
)
创建一个游标对象
cursor = conn.cursor()
3、创建表
在写入数据之前，需要确保目标表已经存在。如果表不存在，可以使用SQL语句创建表。以下是一个创建表的示例：
# 创建表的SQL语句
create_table_query = """
CREATE TABLE IF NOT EXISTS employees (
id INT AUTO_INCREMENT PRIMARY KEY,
name VARCHAR(255) NOT NULL,
salary DECIMAL(10, 2),
department VARCHAR(255)
)
"""
执行创建表的SQL语句
cursor.execute(create_table_query)
4、插入数据
可以使用INSERT INTO SQL语句将数据插入到表中。以下是插入数据的示例：
# 插入数据的SQL语句
insert_query = """
INSERT INTO employees (name, salary, department)
VALUES (%s, %s, %s)
"""
要插入的数据
data = ("John Doe", 50000.00, "Engineering")
执行插入数据的SQL语句
cursor.execute(insert_query, data)
提交事务
conn.commit()
获取插入数据后的行ID
last_id = cursor.lastrowid
print(f"Inserted data with ID: {last_id}")
5、批量插入数据
如果需要插入大量数据，可以使用批量插入的方式来提高效率。以下是批量插入数据的示例：
# 批量插入数据的SQL语句
insert_query = """
INSERT INTO employees (name, salary, department)
VALUES (%s, %s, %s)
"""
要插入的数据列表
data = [
("Alice", 60000.00, "Marketing"),
("Bob", 45000.00, "Sales"),
("Charlie", 55000.00, "HR")
]
执行批量插入数据的SQL语句
cursor.executemany(insert_query, data)
提交事务
conn.commit()
6、关闭连接
在完成所有操作之后，需要关闭游标和数据库连接：
# 关闭游标
cursor.close()
关闭数据库连接
conn.close()
总结
本文介绍了Python写入MySQL数据库的四种方法：使用MySQL Connector、使用SQLAlchemy、使用pymysql、使用MySQLdb。推荐使用MySQL Connector和SQLAlchemy，因为它们性能和兼容性较好，且支持最新的Python 3.x版本。在具体项目中，可以根据需求选择合适的方式进行数据写入操作。
相关问答FAQs：
如何使用Python连接MySQL数据库？
要使用Python连接MySQL数据库，您可以使用mysql-connector-python或PyMySQL等库。安装这些库后，您可以通过以下代码连接到数据库：
import mysql.connector
# 建立连接
connection = mysql.connector.connect(
host='localhost',
user='your_username',
password='your_password',
database='your_database'
)
# 创建游标
cursor = connection.cursor()
确保在连接时提供正确的主机名、用户名、密码和数据库名称。
如何在Python中执行插入操作？
在Python中执行插入操作，可以使用游标的execute()方法。您需要构造一个INSERT语句，并将数据作为参数传递给execute()方法。例如：
sql_insert_query = """INSERT INTO your_table (column1, column2) VALUES (%s, %s)"""
values = (value1, value2)
cursor.execute(sql_insert_query, values)
connection.commit() # 提交更改
执行完插入操作后，记得调用commit()来保存更改。
如何处理Python与MySQL之间的错误？
在Python与MySQL交互时，可能会遇到各种错误，如连接错误或SQL语法错误。可以使用try-except语句捕获这些异常，确保程序的稳定性。例如：
try:
# 尝试连接和执行操作
connection = mysql.connector.connect(...)
cursor.execute(...)
except mysql.connector.Error as err:
print(f"Error: {err}")
finally:
if connection.is_connected():
cursor.close()
connection.close() # 关闭连接
这样，您可以在发生错误时获取错误信息，并确保数据库连接被正确关闭。
