pandas - 在 PandaSQL 中使用用户输入变量

Question

我正在尝试在我拥有的数据帧上使用 pandaSQL，我想知道是否有使用变量的方法或者是否有另一种方法。我要做的是将用户输入设置为变量，然后尝试在 SQL 语句中使用它。我想在输入时显示形状的每个实例。我正在尝试以下内容：

variable1 = input("Enter shape here: ")
print pysqldf("SELECT imageNum FROM df WHERE shape1 = variable1 ")

但到目前为止还没有运气。其他一切都很好，我只是在引入变量时遇到了麻烦。这在 pandaSQL 中是否可行，如果没有，我可以使用哪些解决方法？

score 1 · Accepted Answer

您需要将WHERESQL 查询中的子句在块引号中传递一个参数，即：

从 df 中选择 imageNum，其中 shape1 =“正方形”。

因此，您需要传递一个包含引号的字符串。尝试以下操作：

variable1 = input("在此处输入形状：")

变量1str = "\"" + 变量1 +"\""

print pysqldf("SELECT imageNum FROM df WHERE shape1 = variable1str")

这样，如果您的输入是"square"：

打印变量
1`>正方形

打印变量1str
'>“正方形”

1 回答 1