nonzero
和argwhere
两个就给大家介绍其中数组中的元素信息True
。where
的工作方式与nonzero
您发布的表单相同,但是具有第二种形式:
np.where(mask,a,b)
可以粗略地认为它是条件表达式的numpy“ ufunc”版本:
a[i] if mask[i] else b[i]
(具有适当的广播a
和b
)。
至于有 两 nonzero
和argwhere
,他们是不同的概念。nonzero
构造为返回可用于索引的对象。如果0稀疏,则比创建整个布尔蒙版更轻巧:
mask = a == 0 # entire array of bools
mask = np.nonzero(a)
现在,您可以使用该掩码对其他数组等进行索引。但是,就目前而言,弄清哪些索引对应于0个元素并不是很好。那就是argwhere
进来的地方。