我们知道,在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和列表的长度相差无几,那么先排序再切片的性能会更高一些。

每日留言

说说你看完本文的感受?

或者一句激励自己的话?

(字数不少于15字)

怎么加入刻意学习队伍

点我,看文末彩蛋

留言有啥福利

点我就知道了

想进学习交流群

加微信:jjxksa888

:简说Python

2小时快速掌握Python基础知识要点

完整Python基础知识要点

Python小知识 | 这些技能你不会?(一)

Python小知识 | 这些技能你不会?(二)

Python小知识 | 这些技能你不会?(三)

Python小知识 | 这些技能你不会?(四)

近期推荐阅读:

【1】整理了我开始分享学习笔记到现在超过250篇优质文章,涵盖数据分析、爬虫、机器学习等方面,别再说不知道该从哪开始,实战哪里找了

【2】【终篇】Pandas中文官方文档:基础用法6(含1-5)

觉得不错就点一下“在看”吧 

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐