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

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/
现在，只要说它采用参数序列作为第二个参数就足够了。