查找向量的 N 个最高值答案

【问题标题】：Finding the N highest values of a vector [duplicate]查找向量的 N 个最高值
【发布时间】：2019-01-18 03:35:35
【问题描述】：

如何找到向量的 N 个最大值？假设我有一个向量：

arr = np.array([12.5, 13.6, −9.1, 17.5, 15.3, 10.5])

N = 3

输出应该是：

np.array([13.6,17.5,15.3])

所以 arr 的 3 个最高值保持相同的顺序。

如果 N=4，输出应该是：

np.array([12.5,13.6,17.5,15.3])

【问题讨论】：

【解决方案1】：

您可能喜欢heapq.nlargest() 方法。它允许您使用线性时间获取任何集合中的n 最大元素。 IE。您无需花费O(n*log n) 时间对原始集合进行排序。

import heapq

n = 4
print(np.array(heapq.nlargest(n, arr)))

【讨论】：