SQL where in 子句使用 Pandas 数据框中的列

Question

我有一个包含一列 ID 的 Pandas 数据框。我需要运行另一个 sql 查询，其“WHERE”子句由上述列中的所有 ID 指定。

前任：

df1 = DataFrame({'IDs' : [1,2,3,4,5,6]})

query = """ Select id, SUM(revenue) AS revenue WHERE id IN (***want df1['IDs'] here***) Group by 1"""

df2 = my_database.select_dataframe(query)

Answer 1

将系列转换为字符串

str = ','.join([str(x) for x in df1['IDs'].tolist()])

str
'1,2,3,4,5,6'

然后将其插入查询字符串 -

qry = "Select id, SUM(revenue) AS revenue WHERE id IN (%s) Group by 1" % str

qry
'Select id, SUM(revenue) AS revenue WHERE id IN (1,2,3,4,5,6) Group by 1'