在PandaSQL中使用用户输入变量

时间:2015-04-07 21:44:03

标签: pandas dataframe pandasql

我正在尝试在我拥有的数据框架上使用pandaSQL,并且我想知道是否有使用变量的方法或者是否有另一种方法来执行此操作。我要做的是将用户输入设置为变量,然后尝试在SQL语句中使用它。我想在输入时显示形状的每个实例。我正在尝试以下方面:

variable1 = input("Enter shape here: ")
print pysqldf("SELECT imageNum FROM df WHERE shape1 = variable1 ")
但到目前为止还没有运气。其他一切工作正常,我在引入变量时遇到麻烦。这在pandaSQL中是否可行,如果没有,我可以使用哪些变通方法?

1 个答案:

答案 0 :(得分:1)

您需要在SQL查询中将WHERE子句传递给blockquotes中的参数,即:

  

SELECT imageNum FROM df WHERE shape1 =“square”。

因此,您需要传递包含引号的字符串。请尝试以下方法:

  

variable1 = input(“在这里输入形状:”)

     

variable1str =“\”“+ variable1 +”\“”

     

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

这样,如果您的输入是"square"

  

打印变量1   `>广场

     

print variable1str
  “> “方形”

相关问题