sqlite3 与 pandas
2019-07-30 本文已影响0人
水之心
对于一个 sqlite3 的小白,通过 sqlite3 的表头使用 pandas 来获取 sqlite3 的数据内容(已经知道 sqlite3 的表头,但是 sqlite3 的表头你是通过第三方软件获知的)是十分容易的。但是,使用第三方软件获取 sqlite3 的表头,实在的对不起 Python 了,因为 Python 知道如何获取 sqlite3 的表头。
class SQLDataset:
def __init__(self, sql_name):
'''
sqlite3存在系统表sqlite_master,结构如下:
sqlite_master(
type TEXT, #类型:table-表,index-索引,view-视图
name TEXT, #名称:表名,索引名,视图名
tbl_name TEXT,
rootpage INTEGER,
sql TEXT
)
'''
self.sql_name = Path(sql_name)
@property
def tab_names(self):
'''
获取 sqlite3 数据的全部表名
'''
with sqlite3.connect(self.sql_name) as conn:
cu = conn.cursor() # 获取表名,保存在tab_name列表
cu.execute("select name from sqlite_master where type='table'")
tab_name = cu.fetchall()
return [line[0] for line in tab_name]
def is_null(self):
if self.tab_names == []:
return True
else:
return False
def _sql2pd(self):
# root sqlite3 数据的 path
with sqlite3.connect(self.sql_name) as con:
# read_sql_query和read_sql都能通过SQL语句从数据库文件中获取数据信息
pd_list = [pd.read_sql(
f"SELECT * FROM {tab_name}", con=con) for tab_name in self.tab_names]
return pd_list
调取方法:
sq = SQLDataset(sql_name)
使用 sq.tab_names
来获取全部的表头名称。