在numpy数组中查找包含最大值的行或列

Question

如何在2d numpy数组中找到包含数组范围最大值的行或列？

Answer 1

如果你只需要一个或另一个:

np.argmax(np.max(x, axis=1))

对于列,和

np.argmax(np.max(x, axis=0))

为行.

Answer 2

你可以用np.where(x == np.max(x)).

例如:

>>> x = np.array([[1,2,3],[2,3,4],[1,3,1]])
>>> x
array([[1, 2, 3],
       [2, 3, 4],
       [1, 3, 1]])
>>> np.where(x == np.max(x))
(array([1]), array([2]))

第一个值是行号,第二个数是列号.

Answer 3

您可以使用np.argmax连同np.unravel_index在

x = np.random.random((5,5))
print np.unravel_index(np.argmax(x), x.shape)