numpy.argwhere

numpy.argwhere(a)

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

参数:

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) .

示例

Try it in your browser!

>>> 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]])

Go BackOpen In Tab