python pandas 查询列表中的值

Question

我想用来query()过滤出现在给定列表中的熊猫数据框中的行。类似于这个问题，但我真的更愿意使用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')