我有一个带有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])
但这似乎应该有更好的方法。
答案 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)