该
argmax()方法应该有所帮助。
更新资料
(在阅读评论后)我相信该
argmax()方法也适用于多维数组。链接的文档提供了以下示例:
>>> a = array([[10,50,30],[60,20,40]])>>> maxindex = a.argmax()>>> maxindex3
更新2
(感谢KennyTM的评论)您可以使用
unravel_index(a.argmax(),a.shape)将索引作为元组获取:
>>> from numpy import unravel_index>>> unravel_index(a.argmax(), a.shape)(1, 0)



