Numpy argmax。如何同时计算最大值和索引?
有没有办法一次性得到最大值和它的位置呢?
import numpy as np
a=[0,0,1,0]
maximum=max(a)
index=np.argmax(a)
有没有更快的方法来做到这一点,比如说:
[maximum,index]=function(a)
1 个回答
41
也许像这样的写法会更快...
index = np.argmax(a)
maximum = a[index]