从sqlite到pandas获取信息行的最快/更好的方法

时间:2017-02-08 16:02:49

标签: python sqlite pandas

我在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'])]

0 个答案:

没有答案