【发布时间】: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 循环?
【问题讨论】:
我有一个数据框 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 循环?
【问题讨论】:
你刚刚发音!
df[df.A.isin([2,3])]
A B
1 2 H
2 3 I
因此它选择行 (df[...]),其中 A 的值 (df.A) 在给定列表 (.isin([2,3])) 中
【讨论】: