蟒蛇＆amp; | 码农家园

python & pandas: subset dataframe with value in a list

本问题已经有最佳答案，请猛点这里访问。

我有一个数据框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循环？

你刚刚念出来了！

1
2
3
4

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

所以它选择行(df[...]，其中(df.A的值在给定的列表(.isin([2,3])中)。