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

Question

NumPy具有有效的函数/方法^{}来识别ndarray对象中非零元素的索引。获得do值为零的元素指数的最有效方法是什么？

Answer 1

numpy.where()是我的最爱。

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

Answer 2

有^{}

import numpy as np
arr = np.array([[1,2,3], [0, 1, 0], [7, 0, 2]])
np.argwhere(arr == 0)

将所有找到的索引返回为行：

array([[1, 0],    # Indices of the first zero
       [1, 2],    # Indices of the second zero
       [2, 1]],   # Indices of the third zero
      dtype=int64)

Answer 3

可以使用以下命令搜索任何标量条件：

>>> a = np.asarray([0,1,2,3,4])
>>> a == 0 # or whatver
array([ True, False, False, False, False], dtype=bool)

它将返回数组作为条件的布尔掩码。