numpy.argwhere

numpy.argwhere(a)

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

参数：

a：array_like

输入数据。

返回：

index_array：ndarray

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

也可以看看

where，nonzero

笔记

np.argwhere(a)与np.transpose(np.nonzero(a))相同。

argwhere的输出不适合索引数组。为此，请改用where(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]])