python＆pandas：列表中具有值的子集数据框[重复]

Posted 2023-03-11

【中文标题】python＆pandas：列表中具有值的子集数据框[重复]

【英文标题】：python & pandas: subset dataframe with value in a list [duplicate]

【发布时间】：2016-06-01 11:00:46

【问题描述】：

我有一个数据框 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】：

你刚刚发音！

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

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

【讨论】：

isin() 函数太有用了！谢谢

如果需要检查两列怎么办？