资讯专栏INFORMATION COLUMN

Python每日一练0006

Batkid / 964人阅读

摘要:问题在某个集合中找到最大或最小的个元素解决方案使用模块例如此外,这两个函数都可以接受作为参数,例如输出为讨论根据官方文档对的介绍可以了解到提供了堆数据结构的实现,并且实现方式是小顶堆,也就是说每次的时候取出的是最小的元素首先使用将一个列

问题

在某个集合中找到最大或最小的N个元素

解决方案

使用heapq模块

heapq.nlargest(n, iterable, key=None)

heapq.nsmallest(n, iterable, key=None)

例如:

>>> import heapq
>>> l = [9, -2, 0, 8, 1, 3]
>>> print(heapq.nlargest(2, l))
[9, 8]
>>> print(heapq.nsmallest(2, l))
[-2, 0]

此外,这两个函数都可以接受key作为参数,例如:

import heapq
fruits = [
    {"name": "orange", "price": 5},
    {"name": "apple", "price": 2},
    {"name": "pear", "price": 1.5},
    {"name": "lemon", "price": 3},
]
print(heapq.nlargest(2, fruits, key=lambda x: x["price"]))

输出为:

[{"name": "orange", "price": 5}, {"name": "lemon", "price": 3}]
讨论

根据Python3官方文档对heapq的介绍可以了解到

heapq提供了堆数据结构的实现,并且实现方式是小顶堆,也就是说每次pop的时候取出的是最小的元素

首先使用heapq.heapify将一个列表初始化为堆

>>> import heapq
>>> l = [-1, 2, 5, 0, 8]
>>> heapq.heapify(l)
>>> print(l)
[-1, 0, 5, 2, 8]

然后就可以调用heapq.heappushheapq.heappop对堆进行增加和删除操作了

>>> heapq.heappush(l, 8)
>>> print(l)
[-1, 0, 5, 2, 8, 8]
>>> print(heapq.heappop(l))
-1

此外,heapq还提供了其他堆的一些操作

heapq.heappushpop(heap, item)先将item存入堆中,然后弹出最小的元素,相当于先调用了heapq.heappush(item)再调用heapq.heappop(),但这样调用会比分开调用两个函数效率更高

heapq.heapreplace(heap, item)先弹出最小的元素,再存入item

heapq.merge(*iterables, key=None, reverse=False)将多个有序的集合合并成一个有序的集合,并且返回的是迭代器对象

来源

Python Cookbook

关注

欢迎关注我的微信公众号:python每日一练

文章版权归作者所有,未经允许请勿转载,若此文章存在违规行为,您可以联系管理员删除。

转载请注明本文地址:https://www.ucloud.cn/yun/44679.html

相关文章

  • Python每日一练0003

    摘要:问题在每日一练中介绍了如何一个可迭代对象,使用运算符即可但往往我们遇到的问题是可迭代对象中的数量是不确定的这个时候该如何拿到我们想要的元素,比如我们只需要可迭代对象的第一个或者最后一个元素而已解决方案使用中的运算符例如我们需要拿到一个元组的 问题 在每日一练0001中介绍了如何unpack一个可迭代对象,使用,运算符即可 但往往我们遇到的问题是可迭代对象中的数量是不确定的 这个时候该如...

    happyfish 评论0 收藏0
  • Python每日一练0002

    摘要:问题如何序列化输出元素包含字符串元组的字符串元组好绕举个例子将输出为解决方案容易想到使用函数,但函数要求元素必须都是字符串类型,否则会抛出错误一个比较简单的方法是将给进中,然后再将给进函数,最后指定函数的参数来输出如果想要将结果存储起来,那 问题 如何序列化输出元素包含字符串元组的字符串元组(好绕) 举个例子 >>> zoo1 = (monkey, elephant) >>> zoo2...

    yvonne 评论0 收藏0
  • Python每日一练0024

    摘要:问题如何执行外部命令,如解决方案使用库在之前,使用函数在及之后,使用函数讨论命令的执行默认不需要环境,所以当你使用作为参数时,需要将置位,否则会报错误通常来说对于执行系统命令,我们会想到,但在官方文档中已经建议了使 问题 如何执行外部命令,如ls -l 解决方案 使用subprocess库 在Python 3.5之前,使用subprocess.call()函数 >>> import s...

    wqj97 评论0 收藏0
  • Python每日一练0023

    摘要:问题如何判断一个文件是否存在解决方案这个问题可以分成几类问题如果这里的文件指的是文件或目录,我们可以用方法如果这里的文件指的是普通的文件,我们可以用方法如果这里的文件指的是目录,我们可以用方法并且在之后,可以使用面向对象的方法使用库来判断, 问题 如何判断一个文件是否存在 解决方案 这个问题可以分成几类问题 如果这里的文件指的是文件或目录,我们可以用os.path.exists()方法...

    wuyangnju 评论0 收藏0
  • Python每日一练0020

    摘要:问题你需要执行简单的日期操作,计算两个日期间隔多少天某个日期后的多少天是几月几日转换时间字符串的格式等解决方案使用库中的和类其中类代表一个日期时间,例如年月日点分秒类代表一个日期间隔对于实例,可以直接进行数学运算得到一个实例,也就是两个日 问题 你需要执行简单的日期操作,计算两个日期间隔多少天、某个日期后的多少天是几月几日、转换时间字符串的格式等 解决方案 使用datetime库中的d...

    Berwin 评论0 收藏0

发表评论

0条评论

Batkid

|高级讲师

TA的文章

阅读更多
最新活动
阅读需要支付1元查看
<