使用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()
标签:Python,数据库,pymysql,user,库中,SQL,conn
From: https://www.cnblogs.com/bigleft/p/18018727