python pandas 查询列表中的值

Question

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