在Python中，使用PyMySQL将数据写入数据库表有几个关键步骤：连接数据库、创建游标、执行SQL插入语句、提交事务、关闭连接。 其中，创建游标是一个重要步骤，游标用于执行SQL语句和获取查询结果。下面将详细介绍每个步骤，并提供示例代码。
一、连接数据库
使用PyMySQL连接到MySQL数据库需要提供数据库主机名、用户名、密码和数据库名称等信息。
import pymysql
connection = pymysql.connect(
host='localhost',
user='root',
password='password',
database='test_db'
)
二、创建游标
创建游标对象以执行SQL查询。
cursor = connection.cursor()
三、执行SQL插入语句
使用游标对象的execute方法执行SQL插入语句，将数据插入表中。
sql = "INSERT INTO employees (name, age, department) VALUES (%s, %s, %s)"
values = ("John Doe", 30, "HR")
cursor.execute(sql, values)
四、提交事务
执行插入语句后，需要提交事务以确保更改生效。
connection.commit()
五、关闭连接
最后，关闭游标和数据库连接。
cursor.close()
connection.close()
示例代码
以下是完整的示例代码，将数据插入名为employees的表中。
import pymysql
try:
# 连接数据库
connection = pymysql.connect(
host='localhost',
user='root',
password='password',
database='test_db'
)
cursor = connection.cursor()
# 执行SQL插入语句
sql = "INSERT INTO employees (name, age, department) VALUES (%s, %s, %s)"
values = ("John Doe", 30, "HR")
cursor.execute(sql, values)
# 提交事务
connection.commit()
except pymysql.MySQLError as e:
print(f"Error: {e}")
connection.rollback()
finally:
# 关闭连接
cursor.close()
connection.close()
详细步骤解析
一、连接数据库
连接数据库是所有操作的基础。PyMySQL提供了connect方法来建立与数据库的连接，参数包括主机名、用户名、密码和数据库名等。确保提供正确的连接信息，以成功连接到数据库。
二、创建游标
游标是数据库操作的关键。通过游标，可以执行SQL查询、插入、更新和删除操作。PyMySQL提供了多种游标类型，默认的游标类型是Cursor，它返回结果集的元组形式。
三、执行SQL插入语句
SQL插入语句用于将数据插入到表中。使用游标的execute方法可以执行插入语句。插入语句中使用占位符%s，实际值通过元组传递给execute方法。
四、提交事务
数据库操作需要事务管理。执行插入操作后，必须提交事务以确保更改生效。否则，插入的记录不会保存到数据库中。使用connection.commit方法提交事务。
五、关闭连接
操作完成后，必须关闭游标和数据库连接，以释放资源。使用cursor.close和connection.close方法关闭游标和连接。
错误处理
在数据库操作中，可能会遇到各种错误，如连接错误、SQL语法错误等。使用try-except块捕获和处理错误，确保程序能够正常运行。发生错误时，可以回滚事务，避免数据不一致。
try:
# 数据库操作
...
except pymysql.MySQLError as e:
print(f"Error: {e}")
connection.rollback()
插入多条记录
可以使用executemany方法插入多条记录。executemany方法接受一个SQL语句和一个包含多个记录的列表。
sql = "INSERT INTO employees (name, age, department) VALUES (%s, %s, %s)"
values = [
("John Doe", 30, "HR"),
("Jane Smith", 28, "IT"),
("Mike Johnson", 35, "Finance")
]
cursor.executemany(sql, values)
connection.commit()
使用参数化查询
参数化查询可以防止SQL注入攻击。使用占位符%s和实际值，通过游标的execute方法执行参数化查询。
sql = "INSERT INTO employees (name, age, department) VALUES (%s, %s, %s)"
values = ("John Doe", 30, "HR")
cursor.execute(sql, values)
处理特殊字符
在插入包含特殊字符的数据时，需要进行转义。PyMySQL自动处理转义，确保数据正确插入。
sql = "INSERT INTO employees (name, age, department) VALUES (%s, %s, %s)"
values = ("O'Reilly", 30, "HR")
cursor.execute(sql, values)
插入大数据
插入大数据时，可以使用批量插入或分批插入。批量插入使用executemany方法，分批插入可以在循环中执行多个execute方法。
# 批量插入
values = [(f"Employee {i}", i % 50, "Department") for i in range(1000)]
cursor.executemany(sql, values)
分批插入
for i in range(1000):
values = (f"Employee {i}", i % 50, "Department")
cursor.execute(sql, values)
总结
使用PyMySQL将数据写入MySQL数据库表涉及多个步骤，包括连接数据库、创建游标、执行SQL插入语句、提交事务和关闭连接。通过示例代码和详细解析，可以清晰理解每个步骤的操作。掌握这些操作后，可以灵活应对各种数据库操作需求，如插入多条记录、使用参数化查询、处理特殊字符和插入大数据。
相关问答FAQs：
如何在Python中使用pymysql将数据写入MySQL表？
在Python中使用pymysql写入MySQL表，首先需要安装pymysql库，并通过pip install pymysql命令进行安装。接下来，您需要建立与MySQL数据库的连接，创建一个游标对象，然后使用INSERT语句将数据插入到指定的表中。以下是一个简单的示例代码：
import pymysql
# 连接到数据库
connection = pymysql.connect(host='localhost',
user='your_username',
password='your_password',
database='your_database')
try:
with connection.cursor() as cursor:
sql = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
cursor.execute(sql, ('value1', 'value2'))
connection.commit()
finally:
connection.close()
使用pymysql时，如何处理数据库插入操作的异常？
在进行数据库插入操作时，处理异常是非常重要的。可以使用try-except语句捕获可能出现的异常，例如连接错误或SQL语法错误。这样可以确保程序在遇到问题时不会崩溃，并能够提供相应的错误信息。例如：
try:
# 数据库操作代码
except pymysql.MySQLError as e:
print(f"Error occurred: {e}")
在pymysql中，如何批量插入数据到MySQL表？
批量插入数据可以使用executemany()方法，能够提高插入效率。通过将多条记录放在一个列表中并传递给该方法，可以一次性插入多条数据。示例如下：
data = [
('value1', 'value2'),
('value3', 'value4'),
('value5', 'value6')
]
with connection.cursor() as cursor:
sql = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
cursor.executemany(sql, data)
connection.commit()
通过以上方法，您可以轻松地将数据写入MySQL表中，并且能够有效地处理异常和批量插入操作。
