numpy.argwhere #
- 麻木的。argwhere ( a ) [来源] #
查找按元素分组的非零数组元素的索引。
- 参数:
- 类似数组
输入数据。
- 返回:
- index_array (N, a.ndim) ndarray
非零元素的索引。索引按元素分组。该数组的形状为非零项的数量。
(N, a.ndim)
N
笔记
np.argwhere(a)
与 几乎相同np.transpose(np.nonzero(a))
,但会生成 0D 数组的正确形状的结果。的输出
argwhere
不适合索引数组。为此目的请使用nonzero(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]])