扫码加入数据分析学习群

python连接数据库

#%%
import pymysql
# 插入语句
# 打开数据库连接
db = pymysql.connect(host="localhost", user="root", password="root", database="python",charset="utf8")
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# 数据列表
data = [(1,'张三','18'),
(2,'李四','19'),
(3,'王五','20'),
]
try:
# 执行sql语句，插入多条数据
cursor.executemany("insert into test(id, name, score) values (%s,%s,%s)", data)
# 提交数据
db.commit()
except:
# 发生错误时回滚
db.rollback()

# 关闭数据库连接
db.close()
# ======================================================================================== #
#%%
# 导入pymysql模块
import pymysql
# 查询单个数据
# 连接database,参数1主机名或IP；参数2：用户名；参数3：密码；参数4：数据库名称
conn = pymysql.connect(host="localhost", user="root", password="root", database="python",charset="utf8")
# 创建一个可以执行SQL语句的光标对象
cursor = conn.cursor()
# 定义要执行的SQL语句
sql = """
select * from test
"""
# 执行SQL语句
cursor.execute(sql)
# 查询单条数据
data = cursor.fetchone()
print(data);
# 关闭光标对象
cursor.close()
# 关闭数据库连接
conn.close()