numpy.argwhere

原文：https://docs.scipy.org/doc/numpy/reference/generated/numpy.argwhere.html

译者：飞龙 UsyiyiCN

校对：（虚位以待）

numpy.argwhere(a)[source]

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

参数：	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]])