脾气暴躁的argmax。如何同时计算max和argmax?
问题内容:
有一种方法可以通过一次冲程获得max和argmax吗?
import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
是否有最快的方法可以做到,例如:
[maximum,index]=function(a)
问题答案:
也许像这样的东西更快…
index = np.argmax(a)
max = a[index]