我试图获取 numpy 数组中的前 N 个最大值，如果值相等，则进行随机平局。
我可以按如下方式获取前 N 个最大值(取自 here )，但此代码始终返回第一个“4”(即索引 1)。有没有办法让它在4s中随机选择？
>>> a
array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
>>> ind = np.argpartition(a, -4)[-4:]
>>> ind
array([1, 5, 8, 0])
最佳答案
您可以在排序之前随机化顺序，然后重新应用相同的排列:
In [11]: p = np.random.permutation(len(a))
In [12]: p[np.argpartition(a[p], -4)[-4:]]
Out[12]: array([7, 8, 0, 5])
注意:如果我们再次运行它，我们可能会得到另一个解决方案:
In [13]: p = np.random.permutation(len(a))
In [14]: p[np.argpartition(a[p], -4)[-4:]]
Out[14]: array([1, 8, 0, 5])
关于python - 获取 numpy 数组中 N 个最大值的索引，并随机打破平局，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/29997024/