numpy.argwhere#
- numpy.argwhere(a)[source]#
查找数组元素中非零元素的索引,按元素分组。
- 参数:
- aarray_like
输入数据。
- 返回值:
- index_array(N, a.ndim) ndarray
非零元素的索引。索引按元素分组。此数组的形状为
(N, a.ndim)
,其中N
是非零项目的数量。
注释
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]])