python - 如何让 numpy.argmax 返回所有出现的最大值？

Question

我正在尝试找到一个函数，该函数返回给定列表中所有次出现的最大值。

numpy.argmax然而只返回它找到的第一个匹配项。例如:

from numpy import argmax

list = [7, 6, 5, 7, 6, 7, 6, 6, 6, 4, 5, 6]
winner = argmax(list)

print winner

只给出索引 0。但我希望它给出所有索引:0, 3, 5.

Answer 1

正如 np.argmax 的文档所说:“如果多次出现最大值，则返回与第一次出现相对应的索引。”，所以你将需要另一种策略。

你有一个选择是使用 np.argwhere 结合 np.amax:

>>> import numpy as np
>>> listy = [7, 6, 5, 7, 6, 7, 6, 6, 6, 4, 5, 6]
>>> winner = np.argwhere(listy == np.amax(listy))
>>> print(winner)
 [[0]
  [3]
  [5]]
>>> print(winner.flatten().tolist()) # if you want it as a list
[0, 3, 5]