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

Python 快速获取列表中最大或最小的n个数


我们使用到的方法就是python的基础库heapq堆排序算法
首先使用numpy模块生成随机100个整数的列表,再利用heapq.nlargest获取最大的几位数,利用heapq.nsmallest获取最小的几位数。

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)

执行结果如下:
Python基础库heapq获取最大或最小n个数_第1张图片

你可能感兴趣的:(python初学)