heapq.nlargest函数

函数解释

heapq.nlargest(n, iterable, key=None)

该函数用于从可迭代对象iterable中返回前n个最大的元素。它的参数包括：

• n：要返回的最大元素的数量
• iterable：要从中查找最大元素的可迭代对象
• key：一个函数，用于从每个元素中提取比较键

代码示例

import heapqrequired_topk = [5, 10, 15]
neg_sims = [-0.1, -0.4, -0.3, -0.2, -0.5, -0.6]
top_neg_sims = heapq.nlargest(max(required_topk), neg_sims)

[-0.1, -0.2, -0.3, -0.4, -0.5, -0.6]