python pandas 查询列表中的值

Question

我想使用query()过滤出现在给定列表中的熊猫 dataframe 中的行。 类似于这个问题，但我真的更喜欢使用query()

import pandas as pd
df = pd.DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]})
mylist =[5,3]

我试过了：

df.query('A.isin(mylist)')

Answer 1

你可以试试这个，使用@ ，它允许我们在环境中引用一个变量：

df.query('A in @mylist')

或这个：

df.query('A.isin(@mylist)',engine='python')