提出问题:
如何在某集合里面找出最大或最小的K个元素。
解决思路:
找出最大或最下的K个元素,可以使用Python库中的heapq模块,该模块提供两个函数nlargest()求最大K个和nsmallest()求最小K个。
下面我们举例说明:
import heapq
nums=[12,-9,-3,32,9,56,23,0,11,34]
print(heapq.nlargest(4,nums)) #-->最大的4个
print(heapq.nsmallest(4,nums)) #-->最小的4个
运行结果:
[56, 34, 32, 23]
[-9, -3, 0, 9]
分析下,nlar
2022-02-24 17:59:35
120KB
1