蟒蛇和熊貓：集數據框與值列表中的

Question

我有一個數據幀df = pd.DataFrame({'A':[1,2,3,4],'B':['G','H','I','K']})，我想選擇行，其中A列的值在[2,3]

要做到這一點，我寫了一個簡單for-loop：df.loc[[ e in [2,3] for e in df.A],]

是否有任何內置函數可以做到這一點，而不是使用for-loops？

Answer 1

你剛剛發音！

df[df.A.isin([2,3])] 
    A B 
1 2 H 
2 3 I 

所以它選擇行（df[...]）其中A的（df.A）的值是在給定列表（.isin([2,3])）