我在sqlite中有信息,我使用以下方法导入到pandas中:
def get_data(db, id, limit, offset):
cnx = sqlite3.connect('C:/Users/%s' %(db+'.db'))
a = pd.read_sql_query("SELECT * FROM %s LIMIT %s OFFSET %d" %(id, limit, offset), cnx)
return a
cnx.close
为了获得与数据库中某个代码相等的所有行,我将所有表导入一个pandas数据帧,然后从那里进行搜索。这样可以正常工作,但有更好/更快的方法吗?
id_import = ['id1_SH', 'id2_SH', 'id3_SH']
all_ids = []
for i in id_import:
data = get_data(i, 5, 0)
all_ids.append(data)
all_ids = pd.concat(all_ids)
all_ids
all_ids[all_ids.index.isin(['M4052542', 'M4009722'])]