脾气暴躁的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]