保存最后N个元素,找到最大或者最小的N个元素特殊队列deque,heapq模块--cookbook读书笔记

1. 特殊的队列collections.deque,保存最后N个元素

保存有限的历史记录可算是collections.deque的完美应用场景了。

deque与list的区别是:

deque初始化的时候可以规定队列大小,超过预定大小,新加入的元素会挤掉最早进入的元素;

deque不仅仅支持append、pop实现后进先出,还支持appendleft、popleft对任何一边元素进行操作;

>>> from collections import deque

>>> q = deque(maxlen=3)
>>> q.append(1)
>>> q.append(2)
>>> q.append(3)
>>> q.append(4)
>>> q
deque([2, 3, 4], maxlen=3)
>>> q.pop()
4
>>> q
deque([2, 3], maxlen=3)
>>> q.appendleft(1)
>>> q
deque([1, 2, 3], maxlen=3)

2. heapq 模块的函数应用于队列(找到最大或者最小的N个元素)

我们想在某个集合中找出最大或者最小的N个元素,可以采用heapq模块中,有两个函数------nlargest() 和nsmallest()

heapq提供一部分函数可以实现一些快捷的队列应用:
nlargest() 和nsmallest()获得队列中最大或最小的n个元素;
eapq.heapify(),heappush(),heappop()获取最小元素;
>>> nums = [1, 8, 2, 23, 7, -4, 18, 23, 42, 37, 2]
>>> import heapq
>>> print(heapq.nlargest(3, nums))
[42, 37, 23]
import heapq
protfolio=[
{'name':'IBM','shares':100,'price':91.1},
{'name':'FB','shares':200,'price':21.09},
{'name':'AAPL','shares':50,'price':543.22},
{'name':'HPQ','shares':35,'price':31.75},
{'name':'YHOO','shares':45,'price':16.35},
{'name':'ACME','shares':75,'price':115.65}

    ]
cheap=heapq.nsmallest(3,protfolio,key=lambda s:s['price'])
print(cheap)
expensive=heapq.nlargest(3,protfolio,key=lambda s:s['price'])
print(expensive)


特别指出,nlargest() 和nsmallest()支持第三个参数key=callable,这个key和sorted,max,min用法一样


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值