np.argmax 是否返回第一个最大值?

分享于2022年07月17日 argmax numpy python 问答
【问题标题】:np.argmax 是否返回第一个最大值?(Does np.argmax return the first maximal?)
【发布时间】:2022-07-13 23:31:33
【问题描述】:

np.argmax 是否总是返回 第一个 最大元素的索引?

a = np.array([1, 3, -7, ...]) 是一个具有多个最大元素的一维数组,例如索引 10 20 30 。 Numpy 似乎系统地返回 10 而不是 20 30 ,但是我在 documentation 中找不到有关此行为的信息,我在 source code 中找不到实现细节。有人知道吗?

  • Notes 的文档中它说“如果多次出现最大值,则返回与第一次出现相对应的索引。”。这不是你要找的吗?

【解决方案1】:

是的,这实际上已在 numpy.argmax 文档中明确说明:

备注

如果最大值多次出现,则索引 对应于第一次出现的被返回。