如何将数据框作为参数传递给Python中的SQL查询？

Question

我有一个由一列值组成的数据框，我想将其作为参数传递给执行以下sql查询：

query = "SELECT ValueDate, Value"\
        "FROM Table "\
        "WHERE [ID] in ( ? ) "

所以我尝试了（在很多其他事情中）以下内容：

      df = pd.read_sql_query(query, conn, params=[ df['ID'].values ])
      df = pd.read_sql_query(query, conn, params=[ df['ID'].tolist ])
      df = pd.read_sql_query(query, conn, params=[ list(df['ID'].values) ])
       ...

传递数据帧值的正确方法是什么？

注意：我使用的是Microsoft SQL Server，因此查询需要像我一样进行格式化。

Answer 1

这能满足您的需求吗？

import pandas as pd

your_column = pd.Series([1,2,3,4,5,6,7,8,9])

query = "SELECT ValueDate, Value"\
        "FROM Table "\
        "WHERE [ID] in {}".format(tuple(your_column))

print(query)
# 'SELECT ValueDate, ValueFROM Table WHERE [ID] in (1, 2, 3, 4, 5, 6, 7, 8, 9)'

然后你应该能够查询没有其他参数。

df = pd.read_sql_query(query, conn)

Answer 2

params = tuple(df['ID'].values)
sql = "SELECT COUNT(*) FROM foobar WHERE id IN (%s)" % (",".join(["?"]*len(params)),)
cursor.execute(sql, params)