该argmax()方法应该有所帮助。

更新资料

（在阅读评论后）我相信该argmax()方法也适用于多维数组。链接的文档提供了以下示例：

>>> a = array([[10,50,30],[60,20,40]])
>>> maxindex = a.argmax()
>>> maxindex
3

更新2

（感谢KennyTM的评论）您可以使用unravel_index(a.argmax(), a.shape)将索引作为元组获取：

>>> from numpy import unravel_index
>>> unravel_index(a.argmax(), a.shape)
(1, 0)