python - Python中的sqlite3提取与元组中的值相对应的条目

我正在寻找一个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()

    result.extend(temp_res)

我正在寻找可以用sqlite3命令替换for循环的代码。
我想在此行中的某处粘贴(1,2,4)：

c.execute('''SELECT* FROM my_table WHERE val_1 = ?''', (ind,))

而不是执行for循环。
先感谢您

最佳答案

要替换最后一个for循环，您可以在字符串中建立索引/索引列表，然后只访问数据库一次。
请注意，这是一个分为两个步骤的过程，并且不容易受到SQL injection攻击。

my_query = '''SELECT val_1, val_2
              FROM   my_table
              WHERE  val_1 IN ({:s});'''.format(",".join("?"*len(key_indexes)))
# -> 'SELECT val_1, val_2 FROM my_table WHERE val_1 IN (?,?,?);'
c.execute(myquery, ind).fetchall()

另外：
您没有直接询问这个问题，但是可以将第一个for循环和对execute()的调用简化为对executemany()的单个调用。
您应该测试这两个选项中哪个更快，因为DB-API并未确切指定executemany()的实现方式；而是，各RDBMS的性能可能有所不同。

c.executemany('''INSERT INTO my_table VALUES (?,?);''', my_data)

您可以在此处阅读executemany()：http://www.python.org/dev/peps/pep-0249/
现在，只需说它以第二个参数为参数序列即可。