Python基础库heapq获取最大或最小n个数

import heapq
import numpy as np
#声明一个列表
a = np.random.randint(100, size=100)
print(a)
max_list = heapq.nlargest(5, a)
min_list = heapq.nsmallest(5, a)
print("列表中最大的5位数")
print(max_list)
print("列表中最小的5位数")
print(min_list)

执行结果如下：