美文网首页Metabase 数据分析数据库
用Python操作MySQL数据库(pymysql库)

用Python操作MySQL数据库(pymysql库)

作者: 羋学僧 | 来源:发表于2020-01-09 10:45 被阅读0次

    用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()提交事务。

    学习来源

    相关文章

      网友评论

        本文标题:用Python操作MySQL数据库(pymysql库)

        本文链接:https://www.haomeiwen.com/subject/ekxyactx.html