首页 > 编程语言 > Python操控mysql批量插入数据的实现方法
2020
10-28

Python操控mysql批量插入数据的实现方法

在Python中，通过pymysql模块，编写简短的脚本，即方便快捷地控制MySQL数据库

一、连接数据库

        使用的函数：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 将信息清洗完后插入数据库，但一条条插入势必速度效率跟不上，此时将需要使用批量插入的思维。

# coding:utf-8
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()

到此这篇关于Python操控mysql批量插入数据的实现方法的文章就介绍到这了,更多相关Python mysql批量插入数据内容请搜索自学编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持自学编程网！

编程技巧