python＆pandas：列表中具有值的子数据框

Question

我有一个数据框df = pd.DataFrame({'A':[1,2,3,4],'B':['G','H','I','K']})，我想选择A列的值在[2,3]中的行

为此，我编写了一个简单的for循环： df.loc[[ e in [2,3] for e in df.A],]

是否有任何内置函数可以执行此操作，而不是使用for循环？

Answer 1

你才说出来！

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

因此它选择行（df[...]）中df.A给定列表（.isin([2,3])）中A（）的值