python矩阵中最大值的位置,获取多维NumPy数组中最大值的位置

How can I get get the position (indices) of the largest value in a multi-dimensional NumPy array?

解决方案

The argmax() method should help.

Update

(After reading comment) I believe the argmax() method would work for multi dimensional arrays as well. The linked documentation gives an example of this:

>>> a = array([[10,50,30],[60,20,40]])

>>> maxindex = a.argmax()

>>> maxindex

3

Update 2

(Thanks to KennyTM's comment) You can use unravel_index(a.argmax(), a.shape) to get the index as a tuple:

>>> from numpy import unravel_index

>>> unravel_index(a.argmax(), a.shape)

(1, 0)