本文介绍了获取numpy数组中最大项的位置的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

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

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

推荐答案

方法应该有所帮助。

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

(感谢的评论)你可以使用 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数组中最大项的位置的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!

10-10 21:46