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]

Answer 2

与其他答案相比，如果您使用 this item 会更容易：

>>> import numpy as np
>>> your_list = np.asarray([7, 6, 5, 7, 6, 7, 6, 6, 6, 4, 5, 6])
>>> winners = np.flatnonzero(your_list == np.max(your_list))
>>> winners
array([0, 3, 5])

如果你想要一个列表：

>>> winners.tolist()
[0, 3, 5]

Answer 3

在这种情况下，以下算法以O（n）而不是O（2n）运行（即先使用np.argmax然后使用np.argwhere）：

def allmax(a):
    if len(a) == 0:
        return []
    all_ = [0]
    max_ = a[0]
    for i in range(1, len(a)):
        if a[i] > max_:
            all_ = [i]
            max_ = a[i]
        elif a[i] == max_:
            all_.append(i)
    return all_

Answer 4

更简单......

list [list == np.max（list）]

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

4 个答案: