Numpy argmax。如何同时计算最大值和索引?

41 投票
1 回答
26663 浏览
提问于 2025-05-01 14:01

有没有办法一次性得到最大值和它的位置呢?

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]

撰写回答