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')