python - 创建一个 bool 数组，将 numpy 元素与 None 进行比较

Question

我有一个 dtype=object 的 numpy 数组，我想创建一个 bool 数组来标识哪些元素是 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 元素的 bool 数组？我可以用

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

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

Answer 1

您可以使用 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)