在 a.query function 中使用变量

Question

我正在尝试创建一个 function，它采用列中值的输入名称，然后该值将在 df.query function 中使用。 但是，我无法弄清楚如何使它成为它识别为输入的变量。

这就是我现在所拥有的：

def gettingWeeks(stateAbbr, stateName):

    stateCases = cases.query('state == stateName')

但它不识别 stateName。 有没有办法做到这一点？ 谢谢！

Answer 1

Pandas DataFrame.query方法需要根据其特定语法创建的表达式字符串。 要使用当前命名空间中的变量，您必须在变量名称前使用@符号：

stateCases = cases.query("state == @stateName")

应该可以正常工作。

这是文档。