创建一个布尔数组,将numpy元素与None进行比较

时间:2014-09-25 22:27:19

标签: python arrays python-2.7 numpy

我有一个带有dtype = object的numpy数组,我想创建一个布尔数组来标识哪些元素是None。但看起来None表现得与众不同......

a = np.array(['Duck','Duck','Duck','Goose',None,1,2,3,1,3,None,4])
print a == 'Duck'
print a == 3
print a == None

导致

[ True  True  True False False False False False False False False False]
[False False False False False False False  True False  True False False]
False

是否有“numpythonic”方式来获取None元素的布尔数组?我可以用

np.array([x is None for x in a])

但这似乎应该有更好的方法。

1 个答案:

答案 0 :(得分:12)

您可以使用numpy.equal

In [20]: np.equal(a, None)
Out[20]: 
array([False, False, False, False,  True, False, False, False, False,
       False,  True, False], dtype=bool)