numpy.argwhere#

numpy.argwhere(a)[source]#

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

参数:
aarray_like

输入数据。

返回值:
index_array(N, a.ndim) ndarray

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

参见

where, nonzero

注释

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

argwhere 的输出不适合对数组进行索引。为此目的,请使用 nonzero(a) 代替。

示例

>>> import numpy as np
>>> 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]])