我正在寻找一个sqlite3命令,让我选择一个元组给定的条目,让我用一个示例进行说明:
这是我的数据:
my_data = [(1,8),(2,4),(3,5),(4,7),(5,13)]
我正在尝试提取第一个值为1,2或4的条目;因此,我想要的输出是:
[(1, 8), (2, 4), (4, 7)]
我可以通过下面的代码来实现;但是,我认为我的代码不是最佳的:
import sqlite3
my_data = [(1,8),(2,4),(3,5),(4,7),(5,13)]
key_indexes = (1,2,4)
conn = sqlite3.connect(':memory:')
c = conn.cursor()
c.execute('''CREATE TABLE my_table
(val_1 INTEGER, val_2 INTEGER)''')
for entry in my_data:
c.execute('''INSERT INTO my_table VALUES(?,?)''',entry)
conn.commit()
result = []
for ind in key_indexes:
c.execute('''SELECT* FROM my_table WHERE val_1 = ?''', (ind,))
temp_res = c.fetchall()
r