如何才能获得多维numpy数组中最大项的位置?
解决方法:
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)