pymysql批量插入数据库并跟新

原创

mob649e8168b406 2024-02-12 09:20:43 ©著作权

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

pymysql批量插入数据库并更新

在开发中，我们经常需要向数据库中批量插入数据并进行更新。本文将重点介绍如何使用pymysql库实现批量插入和更新数据库的操作。

1. pymysql简介

pymysql是一个用于Python连接和操作MySQL数据库的库。它遵循Python数据库API v2.0规范，可以方便地进行连接、查询、执行SQL语句等操作。pymysql支持Python 2.7或以上版本，是Python中最常用的MySQL库之一。

2. 批量插入数据

在实际开发中，往往需要一次性向数据库中插入大量数据。使用pymysql可以很方便地实现批量插入操作。

首先，我们需要安装pymysql库。可以使用pip命令进行安装：

pip install pymysql

接下来，我们需要导入pymysql库，并建立与数据库的连接。假设我们已经有一个名为testdb的数据库，其中有一个名为users的表，表结构如下：

id	name	age
1	Alice	25
2	Bob	30
3	Cindy	28

import pymysql

# 建立数据库连接
conn = pymysql.connect(host='localhost', user='root', password='123456', database='testdb')
cursor = conn.cursor()

接下来，我们需要准备要插入的数据。假设我们有一个名为user_data的列表，其中保存了多个用户的信息：

user_data = [
    ('4', 'David', 35),
    ('5', 'Emily', 27),
    ('6', 'Frank', 32)
]

为了实现批量插入，我们可以使用executemany()方法。该方法接受两个参数，第一个参数是执行的SQL语句，第二个参数是要插入的数据。

# 执行批量插入操作
sql = "INSERT INTO users (id, name, age) VALUES (%s, %s, %s)"
cursor.executemany(sql, user_data)

# 提交事务
conn.commit()

# 关闭数据库连接
cursor.close()
conn.close()

通过上述代码，我们成功将user_data列表中的数据批量插入到了数据库的users表中。可以通过查询数据库验证插入结果。

3. 批量更新数据

除了批量插入，我们还可以使用pymysql库实现批量更新数据库的操作。

首先，我们需要建立与数据库的连接，可以参考上述的连接步骤。

接下来，我们需要准备要更新的数据。假设我们要将id为1、2和3的用户年龄都增加2岁：

user_data = [
    (27, 1),
    (32, 2),
    (30, 3)
]

然后，我们可以使用executemany()方法执行批量更新操作。具体的SQL语句可以根据实际需求进行编写。

# 执行批量更新操作
sql = "UPDATE users SET age = %s WHERE id = %s"
cursor.executemany(sql, user_data)

# 提交事务
conn.commit()

# 关闭数据库连接
cursor.close()
conn.close()

通过上述代码，我们成功将id为1、2和3的用户年龄增加了2岁。可以通过查询数据库验证更新结果。