numpy.argwhere #

麻木的。argwhere ( a ) [来源] #

查找按元素分组的非零数组元素的索引。

参数
类似数组

输入数据。

返回
index_array (N, a.ndim) ndarray

非零元素的索引。索引按元素分组。该数组的形状为非零项的数量。(N, a.ndim)N

也可以看看

where,nonzero

笔记

np.argwhere(a)与 几乎相同np.transpose(np.nonzero(a)),但会生成 0D 数组的正确形状的结果。

的输出argwhere不适合索引数组。为此目的请使用nonzero(a)

例子

>>> x = np.arange(6).reshape(2,3)
>>> x
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x>1)
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])