如何将数据框作为参数传递给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)