Question

我正在寻找一个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循环。提前谢谢你

Answer 1

要替换最后一个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/
现在，只需说它将第二个参数作为一系列参数就足够了。

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

1 个答案: