python-cookbook

PYTHON-COOKBOOK

一：数据结构和算法

三个基本的写法：

1. 列表生成式
2. 字典生成式
3. 迭代器生成式

---

1、Heapq

取出最大或最小的几个元素

# 最大或者最小的n个元素

import random
from heapq import nsmallest,nlargest

tt = [random.randint(1,10000) for i in range(1000)]

# 取出的元素较少
%time nsmallest(7,tt)
%time nlargest(7,tt)

# 取出的元素很多
%time sorted(tt)[:7]
%time sorted(tt)[-7:]

# 只取出最大的或者最小的
%time min(tt)
%time max(tt)

优先级队列

import heapq

class PriorityQueue:
    def __init__(self):
        self._queue = []
    def push(self,item,priority):
        heapq.heappush(self._queue,(priority,item))
    def pop(self):
        r = heapq.heappop(self._queue)
        return r 

pp = PriorityQueue()
pp.push('a',3)
pp.push('b',9)
pp.push('c',6)
pp.push('d',3)

for i in range(4):
    print(pp.pop())

---

2、字典

多值字典

from collections import defaultdict

d = defaultdict(