SQL where in 子句使用熊猫数据框中的列答案

【问题标题】：SQL where in clause using column in pandas dataframeSQL where in 子句使用熊猫数据框中的列
【发布时间】：2015-06-26 15:29:49
【问题描述】：

我有一个带有一列 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)

【问题讨论】：

标签： sql pandas

【解决方案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'

【讨论】：

【解决方案2】：

为此，我必须用单引号将列表项括起来。

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

【讨论】：

那么你的 ID 不是整数列。