python - 以变​​量作为列名的 Pandas 查询

Question

传递新的信息应用我了解到这是另一个问题的一部分:Unable to query a local variable in pandas 0.14.0

感谢用户@choldgraf。我以不同的方式应用他在上述链接中的回答。

目标:使用变量作为查询中的列名

失败的例子:

import pandas as pd
fooframe = pd.DataFrame({'Size':['Large', 'Medium', 'Small', 'Tiny'], 'Color':[1, 2, 3, 4]})
myvar = 'Size'
subframe = fooframe.query("myvar == 'Large'")

上面的代码返回 'myvar' 的键错误。

import pandas as pd
fooframe = pd.DataFrame({'Size':['Large', 'Medium', 'Small', 'Tiny'], 'Color':[1, 2, 3, 4]})
myvar = 'Size'
subframe = fooframe.query("@myvar == 'Large'")

上面的代码在查询中的 myvar 之前添加了“@”以将 myvar 引用为局部变量。但是，代码仍然返回错误。

Answer 1

感谢用户@choldgraf。我使用了他在另一篇文章 ( Unable to query a local variable in pandas 0.14.0 ) 中提到的技术，但不是针对列中的值，而是针对列名。

通过将变量插入查询字符串中，可以将变量用作 pandas 查询中的列名，如下所示:

import pandas as pd
fooframe = pd.DataFrame({'Size':['Large', 'Medium', 'Small', 'Tiny'], 'Color':[1, 2, 3, 4]})
myvar = 'Size'
subframe = fooframe.query("`{0}` == 'Large'".format(myvar))

(其中反引号用于括起列名，处理列名中的特殊字符和空格。)