collections模块

在内置数据类型(dict、list、set、tuple)的基础上,collections模块还提供了几个额外的数据类型:Counter、deque、defaultdict、namedtuple和OrderedDict等。

namedtuple: 生成可以使用名字来访问元素内容的tuple

gt;>> from collections import namedtuple
>>> Point = namedtuple('Point', ['x', 'y'])
>>> p = Point(1, 2)
>>> p.x
1
>>> p.y
2
知识兔

deque: 双端队列,可以快速的从另外一侧追加和推出对象

使用list存储数据时,按索引访问元素很快,但是插入和删除元素就很慢了,因为list是线性存储,数据量大的时候,插入和删除效率很低。

deque是为了高效实现插入和删除操作的双向列表,适合用于队列和栈:deque 的操作和list一样

gt;>> from collections import deque
>>> q = deque(['a', 'b', 'c'])
>>> q.append('x')
>>> q.appendleft('y')
>>> q
deque(['y', 'a', 'b', 'c', 'x'])
知识兔

Counter: 计数器,主要用来计数

创建


知识兔
>>> c = Counter('which')>>> c.subtract('witch')  # 使用另一个iterable对象更新>>> c['h']1>>> d = Counter('watch')>>> c.subtract(d)  # 使用另一个Counter对象更新>>> c['a']-1

计数值的访问

计数器的更新(update和subtract)

>>> c = Counter('which')>>> c.update('witch')  # 使用另一个iterable对象更新>>> c['h']3>>> d = Counter('watch')>>> c.update(d)  # 使用另一个Counter对象更新>>> c['h']4
>>> c = Counter("abcdefgab")>>> c["a"]2>>> c["c"]1>>> c["h"]0

键的修改和删除

>>> c = Counter("abcdcba")>>> cCounter({'a': 2, 'c': 2, 'b': 2, 'd': 1})>>> c["b"] = 0>>> cCounter({'a': 2, 'c': 2, 'd': 1, 'b': 0})>>> del c["a"]>>> cCounter({'c': 2, 'b': 2, 'd': 1})

elements()

>>> c = Counter('abracadabra')>>> c.most_common()[('a', 5), ('r', 2), ('b', 2), ('c', 1), ('d', 1)]>>> c.most_common(3)[('a', 5), ('r', 2), ('b', 2)] 
>>> c = Counter(a=4, b=2, c=0, d=-2)>>> list(c.elements())['a', 'a', 'a', 'a', 'b', 'b']

most_common([n])

>>> c = Counter("abcdcba")>>> cCounter({'a': 2, 'c': 2, 'b': 2, 'd': 1})>>> d = c.copy()>>> dCounter({'a': 2, 'c': 2, 'b': 2, 'd': 1})

算术和集合操作

>>> c = Counter(a=3, b=1)>>> d = Counter(a=1, b=2)>>> c + d  # c[x] + d[x]Counter({'a': 4, 'b': 3})>>> c - d  # subtract(只保留正数计数的元素)Counter({'a': 2})>>> c & d  # 交集:  min(c[x], d[x])Counter({'a': 1, 'b': 1})>>> c | d  # 并集:  max(c[x], d[x])Counter({'a': 3, 'b': 2})

其他常用操作

sum(c.values())  # 所有计数的总数c.clear()  # 重置Counter对象,注意不是删除list(c)  # 将c中的键转为列表set(c)  # 将c中的键转为setdict(c)  # 将c中的键值对转为字典c.items()  # 转为(elem, cnt)格式的列表Counter(dict(list_of_pairs))  # 从(elem, cnt)格式的列表转换为Counter类对象c.most_common()[:-n:-1]  # 取出计数最少的n个元素c += Counter()  # 移除0和负值

OrderedDict: 有序字典

使用dict时,Key是无序的。在对dict做迭代时,我们无法确定Key的顺序。

如果要保持Key的顺序,可以用OrderedDict

>>> from collections import OrderedDict>>> d = dict([('a', 1), ('b', 2), ('c', 3)])>>> d # dict的Key是无序的{'a': 1, 'c': 3, 'b': 2}>>> od = OrderedDict([('a', 1), ('b', 2), ('c', 3)])>>> od # OrderedDict的Key是有序的OrderedDict([('a', 1), ('b', 2), ('c', 3)])
>>> od = OrderedDict()>>> od['z'] = 1>>> od['y'] = 2>>> od['x'] = 3>>> od.keys() # 按照插入的Key的顺序返回['z', 'y', 'x']

defaultdict: 带有默认值的字典

values = [11, 22, 33,44,55,66,77,88,99,90]my_dict = {}for value in  values:    if value>66:     my_dict['k1'].append(value)    else:        my_dict['k2'] = [value]
from collections import defaultdictvalues = [11, 22, 33,44,55,66,77,88,99,90]my_dict = defaultdict(list)for value in  values:    if value>66:        my_dict['k1'].append(value)    else:        my_dict['k2'].append(value)
计算机