在 Pandas 查询中使用列表

Question

我正在对 DataFrame 执行查询：

Index Category
1     Foo
2     Bar
3     Cho
4     Foo

我想返回类别为“Foo”或“Bar”的行。当我使用代码时：

df.query("Catergory==['Foo','Bar']")

这工作正常并返回：

Index Category
1     Foo
2     Bar
4     Foo

但是，将来我希望动态更改过滤器，所以我写道：

filter_list=['Foo','Bar']
df.query("Catergory==filter_list")

哪个抛出了错误：

UndefinedVariableError: name 'filter_list' is not defined

我尝试过但没有成功的其他变体是：

df.query("Catergory"==filter_list)
df.query("Catergory=="filter_list)

分别生产：

ValueError: expr must be a string to be evaluated, <class 'bool'> given
SyntaxError: invalid syntax

Answer 1

使用@以参考变量query：

filter_list=['Foo','Bar']

df.query("Category == @filter_list")

输出：

   Index Category
0      1      Foo
1      2      Bar
3      4      Foo

Answer 2

使用isin方法。

df.query('Category.isin(@filter_list)')

Answer 3

尝试这个：

df = pd.DataFrame({'Index':[1,2,3,4],'Category':['Foo','Bar','Cho','Foo']})
filter_list = ['Foo','Bar']

df.query(f'Catergory=={filter_list}')