一、cursor.executemany函数介绍
在Python中操作SQL,我们通常会用到cursor对象, 它是Python DB-API中所有数据库游标的超类,可以实现对数据库操作的各种方法,executemany()方法就是其中之一。
executemany()方法用于执行一条 SQL 语句,该 SQL 语句会执行多次,以实现一次操作多个数据,从而提高SQL的执行效率。和execute()方法不同的是,executemany()方法要求传入的参数是元组嵌套的列表,列表中每个元组代表一组数据,而元组中的每个值则代表一个数据的字段值。简而言之,就是executemany()方法实现了批量插入数据,可以使操作更加高效。
import pymysql
# 连接数据库
conn = pymysql.connect(host='localhost', port=3306, user='root', password='666666', db='test_db')
# 创建游标对象
cursor = conn.cursor()
# 定义 SQL 语句
insert_sql = "INSERT INTO student_info VALUES (%s, %s, %s)"
# 定义数据
data = [
('001', '小明', 18),
('002', '小红', 19),
('003', '小刚', 20)
]
# 批量插入数据
cursor.executemany(insert_sql, data)
# 提交
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()
二、executemany函数使用注意事项
当使用executemany()方法批量插入数据时,需要注意以下事项:
1. SQL语句中需要有占位符,因为executemany()方法会将数据作为参数传递给SQL语句执行;
2. 批量插入的数据必须为元组嵌套的列表形式,因为executemany()方法要将每一个元组作为参数传递给SQL语句;
3. 当批量插入大量数据时,我们可以使用executemany()方法来优化插入效率,但是在插入数据时需要考虑到事务的原子性,及时提交或回滚事务。
三、cursor.executemany与for循环一起使用
有时候我们需要批量插入数据到多个表中,这时候可以使用for循环和executemany()方法一起使用。
import pymysql
# 连接数据库
conn = pymysql.connect(host='localhost', port=3306, user='root', password='666666', db='test_db')
# 创建游标对象
cursor = conn.cursor()
# 定义插入数据函数
def insert_data(table_name, data):
# 定义 SQL 语句
insert_sql = f"INSERT INTO {table_name} VALUES (%s, %s)"
# 批量插入数据
cursor.executemany(insert_sql, data)
# 定义数据
table1_data = [('001', '小明'), ('002', '小红'), ('003', '小刚')]
table2_data = [('001', '语文'), ('002', '数学'), ('003', '英语')]
# 批量插入数据
insert_data('table1', table1_data)
insert_data('table2', table2_data)
# 提交
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()
四、批量删除数据的示例
cursor.executemany()不仅可以用于批量插入数据,还可以用于批量删除数据。通过使用executemany()方法和delete语句,可以快速删除指定id的数据。
import pymysql
# 连接数据库
conn = pymysql.connect(host='localhost', port=3306, user='root', password='666666', db='test_db')
# 创建游标对象
cursor = conn.cursor()
# 定义删除数据函数
def delete_data(ids):
# 定义 SQL 语句
delete_sql = "DELETE FROM student_info WHERE id = %s"
# 批量删除数据
cursor.executemany(delete_sql, ids)
# 定义数据
ids = [(1,), (2,), (3,)]
# 批量删除数据
delete_data(ids)
# 提交
conn.commit()
# 关闭游标
cursor.close()
# 关闭连接
conn.close()
原创文章,作者:KYKYV,如若转载,请注明出处:https://www.506064.com/n/317084.html