【Python】使用Python库中的pymysql执行SQL

使用Python连接数据库

在使用petl执行SQL之前，我们需要先连接到数据库。这里以MySQL为例，我们可以使用pymysql库来连接MySQL数据库。首先，安装pymysql库：

pip install pymysql

然后，编写以下代码来连接MySQL数据库：

import pymysql
from petl import *

# 连接数据库
conn = pymysql.connect(host='localhost', user='root', password='password', db='test', charset='utf8')

执行SQL查询

接下来，我们可以使用petl库来执行SQL查询。首先，需要创建一个DataFrame对象来存储查询结果：

# 执行SQL查询并将结果存储到DataFrame中
df = read_sql('SELECT * FROM user', conn) # 假设user表在数据库中存在

如果需要执行插入、更新或删除操作，可以使用以下方法：

# 插入数据到user表中（注意：这里的数据应该是一个字典列表）
data = [{
    
    'name': '张三', 'age': 18}, {
    
    'name': '李四', 'age': 20}]
insert_into('user', data, conn) # 假设user表在数据库中存在

如果需要更新或删除数据，可以使用以下方法：

# 更新user表中的数据（注意：这里的条件应该是一个布尔表达式）
update_into('user', {
    
    'name': '王五'}, where({
    
    'id': 1}), conn) # 假设user表在数据库中存在，并且有id字段作为主键

关闭数据库连接

最后，不要忘记关闭数据库连接：

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