Python中对sqlite3数据库进行操作时,经常需要用到字段名,然而对于sqlite使用select语句并不能象MySql等数据库一样返回带字段名的字典数据集。特别是对于一个不熟悉的sqlite数据库,写代码时如果需要借助工具查看,那其实是有点对不起python的。
下面两段代码就可以轻易获得一个sqlite数据库中所有数据表名和字段名的列表:
# python 获取sqlite3数据库mydb.db中的表名和表字段名
import sqlite3
conn=sqlite3.connect('mydb.db')
cu=conn.cursor()
#获取表名,保存在tab_name列表
cu.execute("select name from sqlite_master where type='table'")
tab_name=cu.fetchall()
tab_name=[line[0] for line in tab_name]
#获取表的列名(字段名),保存在col_names列表,每个表的字段名集为一个元组
col_names=[]
for line in tab_name:
cu.execute('pragma table_info({})'.format(line))
col_name=cu.fetchall()
col_name=[x[1] for x in col_name]
col_names.append(col_name)
col_name=tuple(col_name)
#之所以保存为元组,一是可避免误操作修改字段名,二是元组巧用转化字符串,可
#直接用于SQL的insert语句中。例如下面代码可得到第一个表的带括号字段名集合:
'''
sql_col_name=str(col_names[0]).replace('\'','')
'''
接下来,用自动获取的字段名做点事吧。
比如,我一个sqlite数据库的表数据是从Excel表中导入的,Excel表中一些误录入的回车符也被一同导入了,在sqlite表中会显示为‘\n'的字符,我想把所有字段中的这个字符都去掉,我是通过以下代码实现的。
#删除第一个表中所有文本型字段中的回车符(注:存入数据库中的回车符取出
#后会展现为原始字符'\n')
cu.execute('select * from '+ tab_name[0])
cdset=cu.fetchall()
cdset=[list(line) for line in cdset] #将结果集的元组转为列表,才能修改。
for line in cdset:
for x in range(len(line)):
#注:作为For循环计数变量不能在循环体中修改,此处不能用x in line代替
if type(line[x])==str and (r'\n' in line[x]): #只能修改字符型数据
line[x]=line[x].replace(r'\n','')
sql="update {bm} set {cm}='{vm}' where {cm2}={vm2}".format(
bm=tab_name[0],cm=col_names[0][x],vm=line[x],cm2=col_names
[0][0],vm2=line[0])
#此处cm2必须是表中具有排它性的主键字段,非字符型字段
cu.execute(sql)
conn.commit()
以上这篇python 获取sqlite3数据库的表名和表字段名的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。