Numpy argmax.如何计算max和argmax?

Pat*_*ceG 21 python numpy

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

import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
Run Code Online (Sandbox Code Playgroud)

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

[maximum,index]=function(a)
Run Code Online (Sandbox Code Playgroud)

Pat*_*ceG 22

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

index=np.argmax(a)
max=a[index]
Run Code Online (Sandbox Code Playgroud)

  • 然而,节省计算时间的好处是,如果原始 argmax 调用要在第二个轴上对二维数组进行排序,则第二个语句可能会引发错误。 (2认同)