脾气暴躁的argmax。如何同时计算max和argmax？

小编典典

python

有一种方法可以通过一次冲程获得max和argmax吗？

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

是否有最快的方法可以做到，例如：

[maximum,index]=function(a)

阅读 212

2021-01-20

共1个答案

小编典典

也许像这样的东西更快…

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

2021-01-20