Python中使用heapq模块找到最大或最小的N个元素

Python中使用heapq模块找到最大或最小的N个元素

在heapq中有两个函数模块，分别是nlargest()、nsmallest(),可以用这两个函数分别找到列表中最大和最小的元素。

这两个函数分别有三个参数，分别为n, iterable, key

1. n: 限制需要的元素个数
2. iterable：可迭代对象
3. key：对复杂的数据进行处理

情况一：仅对列表数据进行操作

对列表进行处理时无需考虑key值，仅需限定参数n、iterable即可。

'''
Author：爱吃跳跳糖的小怪兽
time：2020-8-27
Version：1.0
'''

import heapq

List_Example1 = [-100, 2, 5, 7, -78, 85, 56, -21, 0]     #假定一个无序的整数列表需要进行数据提取
print(heapq.nlargest(3,List_Example1))
print(heapq.nsmallest(3,List_Example1))

Pycharm中运行结果如下：