请参考此答案,该答案还详细说明了如何找到最大值及其(1D)索引,您可以使用argmax()
>>> a = array([[10,50,30],[60,20,40]])
>>> maxindex = a.argmax()
>>> maxindex
3
然后,您可以使用unravel_index(a.argmax(), a.shape)
获取索引作为元组:
>>> from numpy import unravel_index
>>> unravel_index(a.argmax(), a.shape)
(1, 0)