函数解释
heapq.nlargest(n, iterable, key=None)
该函数用于从可迭代对象iterable中返回前n个最大的元素。它的参数包括:
- n:要返回的最大元素的数量
- iterable:要从中查找最大元素的可迭代对象
- key:一个函数,用于从每个元素中提取比较键
代码示例
import heapq
required_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]