获取numpy数组中最大项的位置 [英] Get the position of the biggest item in a numpy array

问题描述

如何获得多维numpy数组中最大项的位置？

How can I get get the position of the biggest item in a multi-dimensional numpy array?

推荐答案

argmax（） 方法应该有所帮助。

The argmax() method should help.

更新

（阅读评论后）我相信 argmax（）方法也适用于多维数组。链接文档提供了一个示例：

(After reading comment) I believe the argmax() method would work for multi dimensional arrays as well. The linked documentation gives an example of this:

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

更新2

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

(Thanks to KennyTM's comment) You can use unravel_index(a.argmax(), a.shape) to get the index as a tuple:

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

这篇关于获取numpy数组中最大项的位置的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！