numpy.argwhere

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

译者:飞龙 UsyiyiCN

校对:(虚位以待)

numpy.argwhere(a)[source]

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

参数:

a:array_like

输入数据。

返回:

index_array:ndarray

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

也可以看看

wherenonzero

笔记

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