python使用pymysql批量插入
时间: 2024-10-15 11:30:40 浏览: 6
在Python中,你可以使用`pymysql`库来连接MySQL数据库并进行批量插入操作,这可以提高数据插入的效率。以下是一个简单的例子:
```python
import pymysql
# 创建连接
db = pymysql.connect(host='localhost', user='your_username', password='your_password', database='your_database')
try:
# 创建游标
cursor = db.cursor(pymysql.cursors.DictCursor)
# 数据列表,假设这是你要插入的数据
data_list = [
{'column1': 'value1', 'column2': 'value2'},
{'column1': 'value3', 'column2': 'value4'},
# 更多数据...
]
# 批量插入SQL语句
batch_query = "INSERT INTO your_table (column1, column2) VALUES (%s, %s)"
cursor.executemany(batch_query, data_list)
# 提交事务
db.commit()
except Exception as e:
print(f"Error occurred: {e}")
# 如果有错误,回滚事务
db.rollback()
finally:
# 关闭游标和连接
cursor.close()
db.close()
```
在这个例子中,我们首先创建一个连接到MySQL的游标,并设置为字典模式以便更方便地处理查询结果。然后定义一个包含待插入数据的列表,使用`executemany()`方法执行批量插入,该方法会将整个列表作为一次发送给数据库。最后别忘了关闭游标和连接。
阅读全文