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

我们知道&＃xff0c;在Python里面&＃xff0c;可以使用 max和 min获得一个列表的最大、最小的元素&＃xff1a;

a &＃61; [4, 2, -1, 8, 100, -67, 25]
max_value &＃61; max(a)
min_value &＃61; min(a)
print(max_value)
print(min_value)


运行效果如下图所示&＃xff1a;

那么问题来了&＃xff0c;如何获取最大的3个元素和最小的5个元素&＃xff1f;

你当然可以先排序&＃xff0c;然后再取&＃xff1a;

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


那有没有其他办法呢&＃xff1f;

答案是使用Python自带的 heapq&＃xff1a;

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


运行效果如下图所示&＃xff1a;

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

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

每日留言

说说你看完本文的感受&＃xff1f;

或者一句激励自己的话&＃xff1f;

&＃xff08;字数不少于15字&＃xff09;

怎么加入刻意学习队伍

点我&＃xff0c;看文末彩蛋

留言有啥福利

点我就知道了

想进学习交流群

加微信&＃xff1a;jjxksa888

备注&＃xff1a;简说Python

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

完整Python基础知识要点

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

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

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

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

近期推荐阅读&＃xff1a;

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

【2】【终篇】Pandas中文官方文档&＃xff1a;基础用法6&＃xff08;含1-5)

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