一日一技:在Python里面如何获取列表的最大n个元素或最小n个元素?

简介: 一日一技:在Python里面如何获取列表的最大n个元素或最小n个元素?

我们知道,在Python里面,可以使用 maxmin获得一个列表的最大、最小的元素:

a = [4, 2, -1, 8, 100, -67, 25]
max_value = max(a)
min_value = min(a)
print(max_value)
print(min_value)

运行效果如下图所示:

那么问题来了,如何获取最大的3个元素和最小的5个元素?

你当然可以先排序,然后再取:

a = [4, 2, -1, 8, 100, -67, 25, 3, 4, 5, 6, 7, 55]
a.sort()
print(f'最小的5个元素:{a[:5]}')
print(f'最大的三个元素:{a[-3:]}')

那有没有其他办法呢?

答案是使用Python自带的 heapq

import heapq
a = [4, 2, -1, 8, 100, -67, 25, 3, 4, 5, 6, 7, 55]
max_three = heapq.nlargest(3, a)
min_five = heapq.nsmallest(5, a)
print(f'最大的3个元素:{max_three}')
print(f'最小的5个元素:{min_five}')

运行效果如下图所示:

这里的 heapq是一个用于处理 这种数据结构的模块。它会把原来的列表转换成一个堆,然后取最大最小值。

需要注意,当你要取的是前n大或者前n小的数据时,如果n相对于列表的长度来说比较小,那么使用 heapq的性能会比较好。但是如果n和列表的长度相差无几,那么先排序再切片的性能会更高一些。

目录
相关文章
|
1天前
|
算法 搜索推荐 Java
【经典算法】LeetCode 215. 数组中的第K个最大元素(Java/C/Python3实现含注释说明,Medium)
【经典算法】LeetCode 215. 数组中的第K个最大元素(Java/C/Python3实现含注释说明,Medium)
6 3
|
2天前
|
索引 Python 容器
6.Python【序列】- 列表
6.Python【序列】- 列表
|
6天前
|
存储 索引 Python
Python列表的循环遍历详解
Python列表的循环遍历详解
7 1
|
6天前
|
索引 Python
Python 列表(List)
Python 列表(List)
|
7天前
|
存储 索引 Python
Python列表
Python列表
|
8天前
|
存储 索引 Python
Python中的列表(List) 详解与高级应用
Python中的列表(List) 详解与高级应用
10 0
|
8天前
|
存储 算法 数据处理
Python中的列表(List) 类型详解与实战应用
Python中的列表(List) 类型详解与实战应用
|
8天前
|
存储 索引 Python
Python列表类型及其操作详解
Python列表类型及其操作详解
|
9天前
|
数据处理 Python
深入理解Python的数据结构:列表与元组
深入理解Python的数据结构:列表与元组
23 1
|
9天前
|
存储 数据处理 索引
Python列表
Python列表