在 NumPy 数组中查找等于 0 的元素索引

NumPy 具有高效的函数/方法nonzero() http://docs.scipy.org/doc/numpy/reference/generated/numpy.nonzero.html识别非零元素的索引ndarray目的。获取元素索引的最有效方法是什么do值为零？

numpy.where() http://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html是我最喜欢的。

>>> x = numpy.array([1,0,2,0,3,0,4,5,6,7,8])
>>> numpy.where(x == 0)[0]
array([1, 3, 5])

方法where返回 ndarray 的元组，每个元组对应于输入的不同维度。由于输入是一维的，因此[0]拆箱元组的唯一元素。