来源：Python爬虫与数据挖掘

作者： Python进阶者

大家好，我是Python进阶者。

前言

我们在进行Python编程的时候，时常要将一些数据保存起来，其中最方便的莫过于保存在文本文件了。但是如果保存的文件太大，用文本文件就不太现实了，毕竟打开都是个问题，这个时候我们需要用到数据库。提到数据库，相信大部分人都不会陌生，今天我们要学的就是数据库中小编自认为最棒的Mysql数据库了。

一、下载导入模块

为了让Python与Mysql 交互，这里我们需要用到Pymsql模块才行。

下载模块：

pip install pymysql

导入模块:

 import pymysql

二、创建数据库

打开数据库连接软件 SqlYong，如图：

输入命令：

CREATE DATABASE IF NOT EXISTS people;

这样就创建了一个people 数据库。

三、创建数据表，并写入数据

USE people;
CREATE TABLE IF NOT EXISTS student(id INT PRIMARY KEY AUTO_INCREMENT,NAME CHAR(10) UNIQUE,score INT NOT NULL,tim DATETIME)ENGINE=INNOBASE CHARSET utf8;
INSERT INTO student(NAME,score,tim)VALUES('fasd',60,'2020-06-01')
SELECT * FROM student;

通过上述操作便创建了一个数据表Student并向其中写入了数据，结果如下：

我们可以一行代码删除这个插入的 数据：

TRUNCATE student;

四、Mysql与Python建立连接

将下图中的参数依次填入初始化参数中，

db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')

这样就连接到了people数据库，可以看下连接成功的打印信息：

可以看到我们打印了Mysql的版本和Host信息。

五、创建游标执行操作

1.创建游标

cur=db.cursor

2.编写插入数据表达式

sql="INSERT INTO student(NAME,score,tim)VALUES('任性的90后boy',100,now())"

3.开启游标事件

cur.begin()

4.执行数据库语句，异常判断

try:
    cur.execute(sql) 执行数据库语句
except Exception as e:
    print(e)
    db.rollback()   发生异常进行游标回滚操作
else:
    db.commit()   提交数据库操作
finally:
    cur.close()  关闭游标
    db.close()  关闭数据库

5，执行插入操作

数据库建立好后，我们可以对它们进行插入数据的操作。

import time
db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')
cur=db.cursor()
db.begin()
sql="INSERT INTO student(NAME,score,tim) VALUES ('%s',%d,'%s')"
data=('HW',90,tt)
try:
  cur.execute(sql%data)
except Exception as e:
  print(e)
  db.rollback()
else:
  db.commit()
finally:
  cur.close()
  db.close()

这样就可以将数据插入进去了。我们还可以自定义插入：

import pymysql
import time
tt=time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')
cur=db.cursor()
db.begin()
s=input('string:')
d=input('number:')
sql="INSERT INTO student(NAME,score,tim)VALUES('%s','%s','%s')"
try:
  data=(s,d,tt)
  cur.execute(sql%data)
except Exception as e:
  print(e)
  db.rollback()
else:
  db.commit()
finally:
  cur.close()
  db.close()

另外，我们也可以同时插入多条数据，只需先定义好所有的数据，然后在调用即可，这里需要用到插入多条数据的函数Executemany，在这里我插入十万条数据，并测试插入时间，步骤如下：

import pymysql
import time
start=time.time()
tt=time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')
cur=db.cursor()
db.begin()
sql="insert into student(NAME,score,tim)values(%s,%s,%s)"
def get():
  ab=[]
  for y in range(1,100000):
    if y>=100:
      data=('user-'+str(y),str(str(float('%.f'%(y%100)))),tt)
    else:
      data=('user-'+str(y),str(y),tt)
    ab.append(data)
  return ab


try:
  data=get()
  cur.executemany(sql,data)
except Exception as e:
  print(e)
  db.rollback()
else:
  db.commit()
finally:
  print('插入数据完毕')
  cur.close()
  db.close()
  end=time.time()
  print('用时：',str(end-start))

6.执行更新操作

有些数据我们觉得它过时了，想更改，就要更新它的数据。

import time
db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')
cur=db.cursor()
db.begin()
sql="update student set name='zjj' where score=100 " 当分数是100分的时候将名字改为zjj
try:
  cur.execute(sql%data)
except Exception as e:
  print(e)
  db.rollback()
else:
  db.commit()
finally:
  cur.close()
  db.close()

7.执行删除操作

有时候一些数据如果对于我们来说没有任何作用了的话了，我们就可以将它删除了，不过这里是删除数据表中的一条记录。

import pymysql
db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')
cur=db.cursor()
db.begin()
sql="delete from student where name='fasd';" 当名字等于‘fasd’的时候删除这个记录
try:
  cur.execute(sql)
except Exception as e:
  print(e)
  db.rollback()
else:
  db.commit()
finally:
  cur.close()
  db.close()

你也可以删除表中所有的数据，只需将Sql语句改为:

sql='TRUNCATE student;'

当然你也可以删除表，但是一般不建议这样做，以免误删：

 DROP TABLE IF EXISTS  student;

8.执行查询操作

有时候我们需要对数据库中的数据进行查询，Python也能轻松帮我们搞定。

import pymysql
import time
tt=time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
db=pymysql.connect(host='localhost',user='root',password='123456',port=3306,db='people')
cur=db.cursor()
db.begin()
sql="select * from student;"
try:
  cur.execute(sql)
  res=cur.fetchall() 查询数据库中的数据
  for y in res:
    print(y) 打印数据库中标的所有数据，以元祖的形式
except Exception as e:
  print(e)
  db.rollback()
else:
  db.commit()
finally:
  cur.close()
  db.close()

六、总结

在我们进行网络爬虫的时候，需要保存大量数据，这个时候数据库就派上用场了，可以更方便而且更快捷保存数据。