如何在 Python 中随机返回列表中最大元素之一的索引？

Question

我的程序计算 3 个不同值出现的次数，并将它们存储在像这样的列表中 classes = [class1,class2,class3]

我想返回最大值的索引，但是这些值经常会重复。如果是这种情况，我想返回一个在最大值之间随机选择的索引。

例如，如果我有classes = [5,2,5]，我想输出 0 或 2。 如果我有classes = [1,1,1]，0,1 和 2 就可以了。 最后，如果我有classes = [1,10,1]，我想输出1。

到目前为止，我已经从其他问题中拼凑出这一点，但它的行为不是我想要的，我不明白为什么。

classes = [class1,class2,class3]
dupes = [n for n, x in enumerate(classes) if x in classes[:n]]
class_prediction = dupes[random.randint(0,len(dupes))] + 1

（最后的+1是返回实际的类标签而不是索引。）

Answer 1

您可以将random.choice 与列表理解一起使用，如下所示

indices = [idx for idx, val in enumerate(values) if val == largest]
random.choice(indices)

举个例子，这里是调用几次的结果

>>> import random
>>> values = [5, 2, 5, 3, 1]
>>> largest = max(values)
>>> indices = [idx for idx, val in enumerate(values) if val == largest]
>>> random.choice(indices)
2
>>> random.choice(indices)
0
>>> random.choice(indices)
2

Answer 2

可能有更好的解决方案，但这里有一个相当“手动”的方法，应该很容易理解：

import random

def get_random_max(classes):
    maximum_indexes = []
    maximum_value = None

    for i, n in enumerate(classes):
        if maximum_value is None or n > maximum_value:
            maximum_value = n
            maximum_indexes = [i + 1]
        elif n == maximum_value:
            maximum_indexes.append(i + 1)

    return random.choice(maximum_indexes)

print(get_random_max([5, 2, 5]))  # prints 1 or 3 with equal probability