python - 如果值在列表 pandas 中，则 bool 向量

Question

很容易在 pandas 中创建 bool 向量来测试诸如此类的值

DF['a'] > 10

但是你怎么写

DF['a'] in list

根据某个列表或其他列表中系列中每个值的成员资格生成 bool 向量？我收到一个值错误。

我知道我可以非常简单地循环访问数据，但是无需这样做是否可以实现？

Answer 1

使用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