码迷,mamicode.com
首页 > 数据库 > 详细

(数据科学学习手札51)用pymysql来操控MySQL数据库

时间:2018-10-26 19:41:51      阅读:207      评论:0      收藏:0      [点我收藏+]

标签:其他   .sh   方便   失败   对象   pytho   navicat   成功   例子   

一、简介

  pymysql是Python中专门用来操控MySQL数据库的模块,通过pymysql,可以编写简短的脚本来方便快捷地操控MySQL数据库,本文就将针对pymysql的基本功能进行介绍;

 

二、操控数据库

2.1 连接数据库

  利用pymysql.connect(host,user,password,port,db)来实现对已知MySQL数据库的连接,其中各参数分别对应着目标数据库的各项属性,db用于指定要连接的database的名称,下面是一个示例:

  要连接的数据库:

技术分享图片

import pymysql

‘‘‘连接数据库‘‘‘
try:
    print(-*200)
    print(尝试连接数据库)
    print(-*200)
    db = pymysql.connect(host=localhost,user=root,password=‘我的密码,port=3306,db=test)
    print(连接成功!,\n,
          host:{}.format(db.host),\n,
          port:{}.format(db.port),\n,
          user:{}.format(db.user),\n,
          database:{}.format(db.db))
    
except Exception as e:
    print(因{},连接失败.format(e))
finally:
    print(-*200,\n,连接过程结束)
    print(-*200)

  若连接成功,显示如下信息:

技术分享图片

 

2.2 插入数据

  在成功连接数据库之后,我们使用db.cursor()来获取数据库的操作游标:

‘‘‘获取操作游标‘‘‘
cur = db.cursor()

  接下来我们使用.execute()来执行需要完成的SQL语句,其传入参数为字符串类型的SQL语句,譬如,下面的例子中我们创建一个新的表,并将sklearn中内置的鸢尾花数据传入进去:

from sklearn.datasets import load_iris

‘‘‘获取鸢尾花数据,其中X,y为二维数组‘‘‘
X,y = load_iris(return_X_y=True)
‘‘‘建立指定鸢尾花类别名称的列表‘‘‘
Species = [setosa,versicolor,virginica]

‘‘‘通过操作游标执行SQL语句,以创建iris表‘‘‘
cur.execute("CREATE TABLE IF NOT EXISTS iris"
            "(Sepal_Length float,"
            "Sepal_Width float,"
            "Petal_Length float,"
            "Petal_Width float,Species char(20)"
            ")")

‘‘‘构造将X,y数据一次性插入iris的SQL语句‘‘‘
BaseSQL = "INSERT INTO iris VALUES"

for i in range(X.shape[0]):
    BaseSQL += "({},{},{},{},‘{}‘),".format(X[i,0],X[i,1],X[i,2],X[i,3],Species[y[i]])

‘‘‘去除末尾多余的逗号‘‘‘
BaseSQL = BaseSQL[:-1]

‘‘‘执行插入鸢尾花数据的SQL语句‘‘‘
cur.execute(BaseSQL)

  运行完上述语句,在已经连接数据库的navicat中查看iris表中的全部数据:

USE test;
SELECT * FROM iris;

技术分享图片

  这是会发现,查询结果只有一张空表,这是因为在执行完插入数据的语句后,并没有提交结果,使用.commit()向数据库提交结果:

‘‘‘提交结果‘‘‘
db.commit()

在navicat中再次查询得到想要的结果:

技术分享图片

 

 

2.3 查询数据

  查询功能是数据库中核心功能之一,查询取数也是数据分析人员在数据库上最常用的操作,在pymysql中想要完成查询取数的过程,要在执行SQL语句之后,对我们的游标对象使用.fetchall()方法来取得对应的查询结果:

‘‘‘查询取数‘‘‘
cur.execute("SELECT Species,COUNT(*) FROM iris GROUP BY Species")

‘‘‘获取查询结果‘‘‘
results = cur.fetchall()

print(results)

技术分享图片

  可以看到,取回的结果为一个规整的tuple对象,可以按照其格式打印出查询结果:

print(Species,|,Counts)
for result in results:
    print(-*20)
    print(result[0],|,result[1])
print(-*20)

技术分享图片

  或者转换为其他格式保存为其他规整的格式以便进一步分析:

import pandas as pd

df = pd.DataFrame(list(results),columns=[Species,Counts])
print(df)

技术分享图片

  而关于其他对数据库的操作(如删除、更新等),与上面类似,只是涉及到更改数据库中数据时,不要忘记commit();

 

2.4 关闭数据库

  在完成需要的操作后,不要忘记断开与数据库间的连接:

db.close()

 

  以上就是本文的全部内容,如有笔误,望指出!

 

(数据科学学习手札51)用pymysql来操控MySQL数据库

标签:其他   .sh   方便   失败   对象   pytho   navicat   成功   例子   

原文地址:https://www.cnblogs.com/feffery/p/9857753.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!