Numpy argmax.如何计算max和argmax? [英] Numpy argmax. How to compute both max and argmax?

问题描述

有没有办法一次性获得 max 和 argmax ?

Is there a way to get max and argmax by one stroke ?

import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)

有没有最快的方法，比如:

Is there a fastest way to do it, with something like:

[maximum,index]=function(a)

推荐答案

也许这样的事情会更快...

Maybe something like this is faster...

index = np.argmax(a)
max = a[index]

这篇关于Numpy argmax.如何计算max和argmax?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！