测试中需要将数据导入到数据库,或者是做模拟数据,excel表是一个很号好的工具
本文记录如何通过python将excel表的数据导入数据库
操作excel文件需要两个包:一个是xlrd 读excel表,一个是xlwt,写excel表
如果电脑里面没有安装这个包,可以用pip install xlrd导入
#导入MySQL包
import pymysql
#导入读excle的包
import xlrd
#打开数据库连接
conn = pymysql.connect(host="localhost", port=3306, user="root", passwd="123456", db="test1")
#创建游标
cursor=conn.cursor()
#打开EXCEL文件
book=xlrd.open_workbook("d:/test/test2.xls")
#阐明excel表的sheet名
sheet=book.sheet_by_name("sh1")
#数据库插入语句
sql="insert into student values(%s,%s,%s)"
#for循环迭代读取excel表中的每一行数据,第一行是列名,所以从第二行开始读取
for i in range(1,sheet.nrows): #sheet.nrows表示总行数,sheet.ncols表示总列数
a=sheet.cell(i,0).value #表中每个单元格是一个cell对象,这里返回cell对象的值 就是单元格中元素的值
b=sheet.cell(i,1).value
c=sheet.cell(i,2).value
param=(a,b,c)
cursor.execute(sql,param)
conn.commit()
cursor.close()
conn.close()