（转）Python Cookbook 1.4 找到最大或最小的 N 个元素

最新推荐文章于 2025-04-09 22:05:54 发布

转载最新推荐文章于 2025-04-09 22:05:54 发布 · 139 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

原文链接：https://blog.csdn.net/weixin_43354181/article/details/89063158

python 专栏收录该内容

48 篇文章

订阅专栏

原创

Python Cookbook 1.4 找到最大或最小的 N 个元素

2019-04-07 00:58:43 vv安的浅唱阅读数 26 更多

分类专栏： python Python Cookbook

本文链接： https://blog.csdn.net/weixin_43354181/article/details/89063158

当我们需要在一个集合，或者列表中找到最大或最小的 N 个元素，这里我们用到的是 heapq 函数。

1、列表
heapq 模块中有两个函数 nlargest 和 nsmallest，可以接受一个数字 n 和列表作为参数，表示取其中最大或最小的 n 个数字。
在这里插入图片描述
返回的是一个已经排好序的列表，包含最大或者最小的几个元素。

2、字典列表
heapq 还可以通过添加 key 参数作用于字典等更复杂的数据结构。
在这里插入图片描述
3、利用 heapq 找出列表中最小的元素
heapq 中有一个函数 heapq.heapify() 可以对列表进行排序
然后利用 heapq.heapop() 函数每次 pop 出来的都是最小的元素：

注意

如果要查找的元素数量和元素总数相比，相对较小，那么可以直接使用 nlargest 和 nsmallest 函数。
如果只是想找一个元素，那么 max() 和 min() 函数则比较适合。
如果要找的元素个数和集合本身差不多大，更快速的方法是先对集合排序，然后做切片操作，比如：

sorted(nums)[:3]
sorted(nums)[-2:]

     
     1
2

                                </div>
            <link href="https://csdnimg.cn/release/phoenix/mdeditor/markdown_views-e9f16cbbc2.css" rel="stylesheet">
                </div>
</article>
<div class="postTime"> 
    <div class="article-bar-bottom">
        <span class="time">
            文章最后发布于: 2019-04-07 00:58:43            </span>
    </div>
</div>