一、连接数据库
使用的函数:pymysql.connect
语法:db=pymysql.connect(host='localhost', user='root', port=3306, password='Your password', db='database_name')
参数说明:
host:MySQL服务器地址
user:用户名
password:MySQL登录密码
port:MySQL服务器端口号
db:需要连接的数据库名
二、创建表
连接完数据库后,我们需要使用db.cursor()获取数据库的操作游标
cur=db.cursor()
接着使用execute()语句即可执行SQL语句。我们以创建学生成绩表来进行演示:
import pymysql
db=pymysql.connect(host='localhost',user='root',password='123456',port=3 306,db='work1')
cur=db.cursor() ##获取游标
sql='create table grade(id varchar(10),name varchar(10),age int,score int,class varchar(10))'
cur.execute(sql)
打开Navicat查看表的情况如下:
学生成绩表创建成功!
三、批量插入数据
创建完表后,接下来就是插入数据…而插入数据又可以分为两种方式:按行插入和一次性,我们将依次介绍。
按行插入
id=['101','102','103','104','105','106','107','108','109','110']
name=['诸葛亮','刘备','周瑜','张飞','关羽','吕布','貂蝉','黄忠','马超','典韦']
age=[28,30,27,26,28,28,24,26,23,18]
score=[89,56,78,60,40,33,90,95,86,82]
class_=['一班','一班','一班','二班','二班','二班','三班','三班','三班','三班']
sql='insert into grade values(%s,%s,%s,%s,%s)'
for i in range(0,len(id)):
cur.execute(sql,(id[i],name[i],age[i],score[i],class_[i]))
db.commit() #一定要提交数据!!不然在数据库中是没有数据显示的
一次性插入
data = ((id[i], name[i], age[i], score[i], class_[i]) for i in range(0, len(id)))
sql = 'insert into grade values(%s, %s, %s, %s, %s)'
cur.executemany(sql, data)
db.commit()
插入数据后,表grade内容如下:
四、更多
根据以上步骤,我们可知只要在execute语句中放入相应的SQL语句,即可实现对数据库的各种操作。而在pymysql中实现查询数据操作后,还需通过对游标对象使用.fetchall()方法来返回查询结果。
查询目标:筛选出各班的平均分并以降序进行排列
sql='select class,avg(score) from grade group by class order by avg(score) desc'
cur.execute(sql)
rst=cur.fetchall()
print(rst)
从输出结果可知,元素是以tuple的方式进行储存的。我们可以转为更多规范的格式查看:
import pandas as pd
df=pd.DataFrame(list(rst),columns=['班级','平均分'])
df
完成各项操作后,不要忘记断开与数据库的连接:
db.close()
附录:Python MySQL 批量插入数据
在现实生活中,经常会使用到Python 将信息清洗完后插入数据库,但一条条插入势必速度效率跟不上,此时将需要使用批量插入函数。
executemany(templet, args):能同时执行多条语句,执行同样多的语句可比execute()快很多,强烈建议执行多条语句时使用executemany
templet:sql模板字符串,例如:insert into table(id,name,age) values(%s,%s,%s)
args:模板字符串中的参数,是一个list,在list中的每一个元素必须是元组! 例如:[(1, 'mike'), (2, 'jordan'), (3, 'james'), (4, 'rose')]
示例:
import pymysql
# 打开数据库连接
db = pymysql.connect(host='localhost', port=3306,
user='username', passwd='password', db='database_name', charset='utf8')
# 使用cursor()方法获取操作游标
cursor = db.cursor()
# SQL 插入语句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, AGE, SEX) VALUES (%s,%s,%s)"
# 一个tuple或者list
T = (('xiaoming', 31, 'boy'), ('hong', 22, 'girl'), ('wang', 90, 'man'))
try:
# 执行sql语句
cursor.executemany(sql, T)
# 提交到数据库执行
db.commit()
except :
# 如果发生错误则回滚
db.rollback()
# 关闭游标
cursor.close()
# 关闭数据库连接
db.close()
相关文章: