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

时间:2017-04-11 19:06:38

标签: python sql pandas dataframe

我有一个由一列值组成的数据框,我想将其作为参数传递给执行以下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,因此查询需要按照我的格式进行格式化。

2 个答案:

答案 0 :(得分:3)

这能满足您的需求吗?

slug

然后你应该能够在没有其他参数的情况下进行查询。

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)'

答案 1 :(得分:0)

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