python – 获取多维numpy数组中最大项的位置

如何才能获得多维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)
上一篇:python – 按整数索引选择一行pandas系列/ dataframe


下一篇:python – pandas loc vs. iloc vs. ix vs. at vs. iat?