NumPy具有有效的函数/方法nonzero()
来标识ndarray
对象中非零元素的索引。获取值为零的元素的索引的最有效方法是什么?
发布于 2011-01-04 05:44:00
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])
发布于 2017-01-19 10:28:20
那就是np.argwhere
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)
发布于 2011-01-04 05:59:07
您可以使用以下命令搜索任何标量条件:
>>> a = np.asarray([0,1,2,3,4])
>>> a == 0 # or whatver
array([ True, False, False, False, False], dtype=bool)
它会将数组作为条件的布尔掩码返回。
https://stackoverflow.com/questions/4588628
复制相似问题