collections标准库
之前Python的第三方库用的一直很舒服，现在突然发现标准库也有collections这样的神器，可以补充list、set、dict以外的应用
1. namedtuple
namedtuple是一个函数，返回一个自定义的tuple，自定义tuple是通过属性而不是索引，使用体验和C的结构体类似
# coding=utf-8
from collections import namedtuple
Point = namedtuple('Point', ('x', 'y', 'color'))
print(Point) # <class '__main__.Point'>
p = Point(1, 2, 'red')
print(p.x, p.y, p.color) # 1 2 red
p2 = p._replace(x=12) # 这里的更新是返回一个新的Point，p本身不变，因为tuple不可变
print(p, p2) # Point(x=1, y=2, color='red') Point(x=12, y=2, color='red')
2. deque
deque特别适合用于队列和栈操作。虽然list也可以模拟队列和栈，但是list是线性存储的单链表数据结构，适合按索引访问，但插入和删除的效率就比较低，而·基于双向链表的deque特别适合做这些操作
# coding=utf-8
from collections import deque
q = deque(['a', 'b', 'c'])
q.append('x') # 尾部添加
q.appendleft('y') # 头部添加
print(q) # deque(['y', 'a', 'b', 'c', 'x'])
print(q.pop()) # x 尾部弹出
print(q.popleft()) # y 头部弹出
print(q) # deque(['a', 'b', 'c'])
q.insert(1, 'insert') # 插入
q.remove('b') # 删除指定元素
print(q) # deque(['a', 'insert', 'c'])
print(q[1]) # insert 索引
# for循环遍历
for i, e in enumerate(q, start=1):
print(i, ':', e)
# 1 : a
# 2 : insert
# 3 : c
3.defaultdict
普通的dict结构在访问key不存在的对象时会报error，defaultdict可以通过设定默认值解决这个问题，个人感觉在写状态机的时候，初始状态用defaultdict特别方便，有点像C语言的static 变量，只初始一次
# coding=utf-8
from collections import defaultdict
dd = defaultdict(lambda: 10, {}) # 构造函数里的{}可以初始化dd的字典内容
dd['a'] = 10
dd['b'] += 100
print(dd) # defaultdict(<function <lambda> at 0x00000000052C4D08>, {'a': 10, 'b': 110})
# 字符统计的例子
def zero():
return 0
# 生成随机数字串
def generate_digits(length=15):
import string
import random
return random.sample(length * string.digits, length)
chs = generate_digits(5)
stat = defaultdict(zero)
for ch in chs:
stat[ch] += 1
for k, v in stat.items():
print(k, ':', v)
# 2 : 2
# 3 : 1
# 9 : 1
# 5 : 1
4. OrderedDict
普通的dict是无须的，也就是在遍历的时候是无能确定key的顺序的，OrderDict会根据插入顺序排序，可以用在自己封装的对象中，如pytorch中的torch.nn.Sequence就利用了这个特性
# coding=utf-8
from collections import OrderedDict
od = OrderedDict([
('a', 1),
('b', 2),
('c', 3),
])
od['hello'] = 0
print(od) # OrderedDict([('a', 1), ('b', 2), ('c', 3), ('hello', 0)])
for k, v in od.items():
print(k, ':', v)
# a : 1
# b : 2
# c : 3
# hello : 0
5. ChainMap
ChainMap可以将多个dict合并，联合查找
# coding=utf-8
from collections import ChainMap
a = {'a': 1, 'b': 2}
b = {'b': 3, 'd': 4} # 注意这里的b重复
cm = ChainMap(a, b)
print(cm) # ChainMap({'a': 1, 'b': 2}, {'b': 3, 'd': 4})
print(cm['b']) # 2 取第一个
print(cm['d']) # 4
cm['b'] = -2 # 这里修改的也是第一个b，由于是传引用，会修改a中的值
print(a, b) # {'a': 1, 'b': -2} {'b': 3, 'd': 4}
6. Counter
就和名称一样，Counter是个简单的计数器
# coding=utf-8
from collections import Counter
c = Counter('helloworld') # 统计可迭代对象的个数，并自动排序
print(c.most_common(2)) # [('l', 3), ('o', 2)]
for k, v in c.items():
print(k, ':', v)
# h : 1
# e : 1
# l : 3
# o : 2
# w : 1
# r : 1
# d : 1