如何从numpy 2d数组中找到大于特殊数字的元素索引？

Question

我想从numpy 2d数组中找到大于2的元素索引。

像这样

import numpy as np
a = np.array([[1,2,3],[4,5,6]])

#  find indices of element that bigger than 2
# result  = [[0,2],[[1,0],[1,1],[1,2]]

Answer 1

您可以使用In [6]: np.where(a>2) Out[6]: (array([0, 1, 1, 1]), array([2, 0, 1, 2])) ，它将为您提供元组模式（单独的轴）中的预期索引：

np.argwhere()

或直接In [5]: np.argwhere(a>2) Out[5]: array([[0, 2], [1, 0], [1, 1], [1, 2]]) ：

{{1}}