Bool vector如果列表pandas中的值

时间:2014-05-02 03:24:14

标签: python python-2.7 pandas

很容易在pandas中创建一个bool矢量来测试值,例如

DF['a'] > 10

但你怎么写

DF['a'] in list

根据某些列表中的系列中每个值的成员资格生成bool向量?我收到了一个值错误。

我知道我可以非常简单地循环访问数据,但是这可能不必这样做吗?

1 个答案:

答案 0 :(得分:2)

使用isin方法:

DF['a'].isin(list)

示例:

DF = pd.DataFrame({'a':np.arange(5),'b':np.arange(5)*2})
print DF

   a  b
0  0  0
1  1  2
2  2  4
3  3  6
4  4  8

print DF['a'].isin([0,2,3])

0     True
1    False
2     True
3     True
4    False