一日一技:在Python里面如何获取列表的最大n个元素或最小n个元素? |
您所在的位置:网站首页 › python获取list的元素个数函数 › 一日一技:在Python里面如何获取列表的最大n个元素或最小n个元素? |
我们知道,在Python里面,可以使用 max和 min获得一个列表的最大、最小的元素: 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) 觉得不错就点一下“在看”吧 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |