Numpy argmax。如何计算max和argmax?

时间:2014-11-04 15:35:02

标签: python numpy

有没有办法让max和argmax一举?

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

是否有最快的方法,例如:

[maximum,index]=function(a)

1 个答案:

答案 0 :(得分:21)

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

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