用Python操作MySQL数据库(pymysql库)
一、安装pymysql库
如果你想要使用python操作MySQL数据库,就必须先要安装pymysql库,这个库的安装很简单,直接使用:
pip install pymysql
假如上面这种方式还是安装不上,就用如下链接找一个合适的安装包,进行安装。
二、使用python连接mysql数据库
1)六个常用的连接参数
- 参数host:mysql服务器所在的主机的ip;
- 参数user:用户名
- 参数密码:密码
- 参数port:连接的mysql主机的端口,交替是3306
- 参数db:连接的数据库名
-
参数的字符集:当读取数据出现中文会乱码的时候,需要我们设置一下编码;python3默认采用的utf8字符集.
2)python连接mysql的语法
import pymysql
db=pymysql.connect(host='localhost',user='root',password='123456',
port=3306,db='spiders',charset=' utf8')
最基本的参数是主机,用户,密码和端口,必须要有。剩下的两个参数根据你自己的情况决定是否使用。
host指的是mysql服务器安装在哪里,本机安装可以写localhost,也可以写成主机名或者主机ip。
db指的是你要操作的是哪个一个数据库,在进行数据库连接的时候,最好加上这个参数。
3)一个简单的热身案例
# 导包
import pymysql
# 使用pymysql连接上mysql数据库服务器,创建了一个数据库对象;
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
# 开启mysql的游标功能,创建一个游标对象;
cursor = db.cursor()
# 要执行的SQL语句;
sql = "select * from student"
# 使用游标对象执行SQL语句;
cursor.execute(sql)
# 使用fetchone()方法,获取返回的结果,但是需要用变量保存返回结果;
data = cursor.fetchone()
print(data)
# 断开数据库的连接,释放资源;
db.close()
结果如下:
三、cursor游标对象的一些常用方法
1)光标用于执行命令的方法
execute(query,args):执行单条sql语句,接收的参数为sql语句本身和使用的参数列表,返回被引用的行数;
executemany(query,args):执行单条sql语句,但是重复执行参数列表里的参数,返回被引用的行数;
2)光标用于接收返回值的方法
fetchone():返回一条结果行;
fetchmany(size):接收大小条返回结果行。如果size的值大于返回的结果行的数量,则返回cursor.arraysize条数据;
fetchall():接收全部的返回结果行;
四、创建表(建)
import pymysql
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
# 创建一个游标对象;
cursor = db.cursor()
# 建表语句;
sql = """create table person(
id int auto_increment primary key not null,
name varchar(10) not null,
age int not null) charset=utf8"""
# 执行sql语句;
cursor.execute(sql)
# 断开数据库的连接;
db.close()
注意:你在mysql中sql语句怎么写,在这里就怎么写。还有一个细节需要注意的是,在python中,将代码进行多次换行的时候,最好使用三引号
。
五、查询数据(查)
1)fetchone():一次获取一条记录
import pymysql
db = pymysql.connect(host='localhost',user='root',db='huangwei',
password='123456',port=3306,charset='utf8')
cursor = db.cursor()
cursor.execute('select count(*) from person')
aa = cursor.fetchone()
print(aa)
# 注意这一句一定是在循环之外,不能放到循环里面。
# 想想这是为什么?
cursor.execute('select name,age from person')
for i in range(aa[0]):
a,b = cursor.fetchone()
c = "我的名字叫{},今年{}岁".format(a,b)
display(c)
db.close()
结果如下:
2)fetchall():一次获取所有记录
import pymysql
db = pymysql.connect(host='localhost',user='root',db='huangwei',
password='123456',port=3306,charset='utf8')
cursor = db.cursor()
cursor.execute('select name,age from person')
aa = cursor.fetchall()
# print(aa)
for a,b in aa:
c = "我的名字叫{},今年{}岁".format(a,b)
display(c)
db.close()
结果如下:
注:还有一个fetchmany()方法,用于一次性获取指定条数的记录,请自行继续研究。
3)使用pandas中的read_sql()方法,将提取到的数据直接转换为DataFrame,进行操作
import pymysql
import pandas as pd
db = pymysql.connect(host='localhost',user='root',db='huangwei',
password='123456',port=3306,charset='utf8')
cursor = db.cursor()
df1 = pd.read_sql("select * from student where ssex='男'",db)
display(df1)
df2 = pd.read_sql("select * from student where ssex='女'",db)
display(df2)
结果如下:
六、插入数据(增)
1)一次性插入一条数据
import pymysql
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
cursor = db.cursor()
# mysql中SQL语句怎么写,这里就怎么写;
name = "猪八戒"
age = 8000
sql = 'insert into person(name,age) values ("猪八戒",8000)'
try:
cursor.execute(sql)
db.commit()
print("插入成功")
except:
print("插入失败")
db.rollback()
db.close()
1.1)一次性插入一条数据
import pymysql
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
cursor = db.cursor()
# 插入数据
sql = 'insert into person(name,age) values(%s,%s)'
try:
cursor.execute(sql,('孙悟空',100000))
db.commit()
print("插入成功")
except:
print("插入失败")
db.rollback()
db.close()
2)一次性插入多条数据
import pymysql
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
cursor = db.cursor()
# 插入数据
sql = 'insert into person(name,age) values(%s,%s)'
# 注意:(('牛魔王',9000),('铁扇公主',8000),('玉皇大帝',6000))也可以
# 小括号都可以换为中括号
datas = [('牛魔王',9000),('铁扇公主',8000),('玉皇大帝',6000)]
try:
cursor.executemany(sql,datas)
db.commit()
print("插入成功")
except:
print("插入失败")
db.rollback()
db.close()
七、更新数据(改)
import pymysql
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
cursor = db.cursor()
# 更新数据
sql = 'update person set age=%s where name=%s'
try:
cursor.execute(sql,[90000,"玉皇大帝"])
db.commit()
print("更新成功")
except:
print("更新失败")
db.rollback()
db.close()
八、删除数据(删除)
import pymysql
db=pymysql.connect(host='localhost',user='root', password='123456',
port=3306, db='huangwei', charset='utf8')
cursor = db.cursor()
# 删除数据
sql = 'delete from person where age=8000'
try:
cursor.execute(sql)
db.commit()
print("删除成功")
except:
print("删除失败")
db.rollback()
db.close()
总结如下:
①pymysql模块是重置开启mysql的事务功能的,因此,进行“增”,“删”,“改”的时候,一定要使用db.commit()提交事务,否则就看不见所插入的数据。
②进行“增”,“删”,“改”的时候,一定要使用try ... except ...语句,因为万一没插入成功,其余代码都无法执行。当语句执行不成功,我们就db.rollback( )回滚到操作之前的状态;当语句执行成功,我们就db.commit()提交事务。