根据一位博主文章修改,原文地址放在代码里面了。
主要完善两方面,一个是修改个别地方,让python3.6.5适配,可以运行。再就是解决了,读取excel后,抓取数据,出现多余的空格,还有符号,导致和表头不能对应。需要对字符串切片。代码里面有专门注释。一定注意,这个代码就在那里容易报错。
具体运行,要根据自己的excel文件调整,重点就是那几句字符切片的代码,根据报错信息调整一下参数。
点击原来博主的文章地址
import sqlite3
import xlrd
#原文链接:https://blog.youkuaiyun.com/jinmie0193/article/details/96130678
# 参考链接
# [Python初接触:SQLite和Excel操作](https://blog.youkuaiyun.com/chlk118/article/details/52702396)
# [PYTHON 连接和创建SQLite数据库](https://blog.youkuaiyun.com/ANXIN997483092/article/details/79774158)
# [Python的可变长参数](https://www.cnblogs.com/QLeelulu/archive/2009/09/09/1563148.html)
# [python获取Excel数据](https://www.cnblogs.com/mxhmxh/p/9367680.html)
# [python读取excel中单元格的内容返回的5种类型](https://www.bbsmax.com/A/Ae5RD8jNzQ/)
# [python中判断输入是否为数字(包括浮点数)](https://www.cnblogs.com/zxmbky/p/9160822.html)
class ExcelToSqlite(object):
exe = " 执行: "
output = " 输出: "
sheetDataStartIndex = 1 # 数据开始计算的行数,如第0行是表头,第1行及之后是数据
def __init__(self, dbName):
print("初始化数据库实例")
super(ExcelToSqlite, self).__init__()
self.conn = sqlite3.connect(dbName)
self.cursor = self.conn.cursor()
def __del__(self):
print("释放数据库实例")
self.cursor.close()
self.conn.close()
def ExcelToDb(self, excelName, sheetIndex, tableName):
"""
excel转化为sqlite数据库表
:param excelName:excel名
:param sheetIndex:excel中sheet位置
:param tableName:数据库表名
"""
print("Excel文件 转 db")
self.tableName = tableName
excel = xlrd.open_workbook(excelName)
sheet = excel.sheets()[sheetIndex] # sheets 索引
self.sheetRows = sheet.nrows # excel 行数
self.sheetCols = sheet.ncols # excle 列数
fieldNames = sheet.row_values(0) # 得到表头字段名
# 创建表
fieldTypes = ""
for index in range(fieldNames.__len__()):
if (index != fieldNames.__len__() - 1):
fieldTypes += fieldNames[index] + " text,"
else:
fieldTypes += fieldNames[index] + " text"
self.__CreateTable(tableName, fieldTypes)
# 插入数据
for rowId in range(self.sheetDataStartIndex, self.sheetRows):
fieldValues = sheet.row_values(rowId)
self.__Insert(fieldNames, fieldValues)
def __CreateTable(self, tableName, field):
"""
创建表
:param tableName: 表名
:param field: 字段名及类型
:return:
"""
print("创建表 " + tableName)
sql = 'create table if not exists %s(%s)' % (self.tableName, field[:-12]) # 这里注意要切片,插入的表头格式有多余逗号出现
print(self.exe + sql)
self.cursor.execute(sql)
self.conn.commit()
def __Insert(self, fieldNames, fieldValues):
"""
插入数据
:param fieldNames: 字段list
:param fieldValues: 值list
"""
# 通过fieldNames解析出字段名
names = "" # 字段名,用于插入数据
nameTypes = "" # 字段名及字段类型,用于创建表
for index in range(fieldNames.__len__()):
if (index != fieldNames.__len__() - 1):
names += fieldNames[index] +","
nameTypes += fieldNames[index] + " ,text"
else:
names += fieldNames[index]
nameTypes += fieldNames[index] + " text"
# 通过fieldValues解析出字段对应的值
values = ""
for index in range(fieldValues.__len__()-2): # 读取的excel取值,注意循环次数,也要调整不然会出现多余的,空值
cell_value = str((fieldValues[index]))
if (isinstance(fieldValues[index], float)):
cell_value = str((int)(fieldValues[index])) # 读取的excel数据会自动变为浮点型,这里转化为文本
if (index != fieldValues.__len__() - 1):
values += "\'" + cell_value + "\',"
else:
values += "\'" + cell_value + "\'"
# 将fieldValues解析出的值插入数据库
sql = 'insert into %s (%s) values(%s)' % (self.tableName, names[:-2], values[:-1]) # 读取的excel千万注意要切片,否则会多出没用的符号
print(self.exe + sql)
self.cursor.execute(sql)
self.conn.commit()
def Query(self, tableName):
"""
查询数据库表中的数据
:param tableName:表名
"""
print("查询表 " + tableName)
sql = 'select * from %s' % (tableName)
print(self.exe + sql)
self.cursor.execute(sql)
results = self.cursor.fetchall() # 获取所有记录列表
index = 0
for row in results:
print(self.output + "index=" + index.__str__() + " detail=" + str(row)) # 打印结果
index += 1
print(self.output + "共计" + results.__len__().__str__() + "条数据")
def executeSqlCommand(self, sqlCommand):
"""
执行输入的sql命令
:param sqlCommand: sql命令
"""
print("执行自定义sql " + tableName)
print(self.exe + sqlCommand)
self.cursor.execute(sqlCommand)
results = self.cursor.fetchall()
print(self.output + str(results))
for index in range(0, results.__len__()):
print(self.output + str(results[index]))
self.conn.commit()
sqlite3.connect('test333.db') # 自己利用sqlit3创建一个database
dbName = "test333.db" # 把数据库名赋值给函数变量
tableName = "student1" # 数据库 表 名,表存不存在都可以,赋值后,代码会自动创建这个表。
excelName = "C:\\Users\\mayn\\Desktop\\月考成绩_理科.xlsm" # excel名(可加路径)
es = ExcelToSqlite(dbName)
es.ExcelToDb(excelName, 0, tableName)
es.Query(tableName)
#es.executeSqlCommand("select * from " + tableName)
es.executeSqlCommand("select 姓名 from "+tableName+" where 班级=346" )
#原文链接:https://blog.youkuaiyun.com/jinmie0193/article/details/96130678
#以下属于测试
#import sqlite3
#import xlrd
#conn = sqlite3.connect('chengji.db')
#cur=conn.cursor()
#cur.execute('select 姓名 , 总分分数, 总分班名 from allstu00 where 班级=346')
#results =cur.fetchall()
#print(results)
#支持创作者
#再强调一遍,原文链接:
https://blog.youkuaiyun.com/jinmie0193/article/details/96130678
下面是运行状态