Python 中的collections 模組

標籤：collections deque counter

這個模組中實現了一些類，非常靈活。可以用於替代python 內建的dict 、list 、tuple 、set 類型。並且一些功能是這些內建類型所不存在的。

在網路上找了一些資料,重點說說collections 模組中的 deque 、defaultdict、Counter 類

1、class deque
類似於python 內建的 list ，不過它是一個雙向的list。可以在任意一頭進行操作

help(collections.deque)

class deque(__builtin__.object)
  deque([iterable[, maxlen]]) --> deque object

deque 的參數應該時一個iterable.

Example:

In [1]: import collectionsIn [2]: d = collections.deque(‘abcdef‘)In [3]: print ddeque([‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘])In [4]: d.append(‘mm‘)         #追加元素In [5]: print ddeque([‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘, ‘mm‘])In [6]: d.appendleft(‘gg‘)   #從左追加元素In [7]: print ddeque([‘gg‘, ‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘, ‘mm‘])In [8]: d.pop()    # pop 一個元素Out[8]: ‘mm‘In [9]: print ddeque([‘gg‘, ‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘])In [10]: d.popleft() #從左pop 一個元素Out[10]: ‘gg‘In [11]: print ddeque([‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘])In [12]: d.extend(‘xyz‘)   #擴充In [13]: print ddeque([‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘, ‘x‘, ‘y‘, ‘z‘])In [14]: d.extendleft(‘123‘)  #左擴充In [15]: print ddeque([‘3‘, ‘2‘, ‘1‘, ‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘, ‘f‘, ‘x‘, ‘y‘, ‘z‘])In [16]: d.reverse() #反轉In [17]: print ddeque([‘z‘, ‘y‘, ‘x‘, ‘f‘, ‘e‘, ‘d‘, ‘c‘, ‘b‘, ‘a‘, ‘1‘, ‘2‘, ‘3‘])In [22]: d.extend(‘1111‘)In [23]: d.count(‘1‘)      #統計一個元素的個數Out[23]: 5In [24]: print ddeque([‘z‘, ‘y‘, ‘x‘, ‘f‘, ‘e‘, ‘d‘, ‘c‘, ‘b‘, ‘a‘, ‘1‘, ‘2‘, ‘3‘, ‘1‘, ‘1‘, ‘1‘, ‘1‘])In [27]: d.rotate(2)  #右旋轉In [28]: print ddeque([‘1‘, ‘1‘, ‘z‘, ‘y‘, ‘x‘, ‘f‘, ‘e‘, ‘d‘, ‘c‘, ‘b‘, ‘a‘, ‘1‘, ‘2‘, ‘3‘, ‘1‘, ‘1‘])In [29]: d.rotate(-2) #左旋轉In [30]: print ddeque([‘z‘, ‘y‘, ‘x‘, ‘f‘, ‘e‘, ‘d‘, ‘c‘, ‘b‘, ‘a‘, ‘1‘, ‘2‘, ‘3‘, ‘1‘, ‘1‘, ‘1‘, ‘1‘])In [31]: d.clear() #清空In [32]: print ddeque([])

另外在deque 函數中發現一個maxlen參數。主要是限制這個雙向list 的 大小

In [34]: d = collections.deque(xrange(10),5)In [35]: print ddeque([5, 6, 7, 8, 9], maxlen=5)In [36]:

2、class defaultdict

python 內建的dict 類型有一個setdefault 方法，這個方法的意思是。擷取一個dict 中key 對應的value 。若這個key 不存在於dict 中，則在這個dict 設定 key, 這個key 對應的 value 為 defaultdict 設定的相應值

Example:

In [46]: d = {‘Name‘:‘God‘,‘Age‘:28}In [52]: d.setdefault(‘Name‘)   # 這個key 存在於dict 中，因此返回了對應的valueOut[52]: ‘God‘In [48]: d.setdefault(‘test‘)   # 這個key不存在於dict 中，因此它被set 到了dict中，沒有指定value ，則value 為 NoneIn [49]: dOut[49]: {‘Age‘: 28, ‘Name‘: ‘God‘, ‘test‘: None}In [50]: d.setdefault(‘test2‘,‘test2value‘)  # 這個key不存在於dict 中，因此它被set 到了dict中，指定value 為 "test2value" Out[50]: ‘test2value‘In [51]: dOut[51]: {‘Age‘: 28, ‘Name‘: ‘God‘, ‘test‘: None, ‘test2‘: ‘test2value‘}

而class defaultdict類似於dict 中的 setdefault函數. 只是它在初始化這個dict 的時候就已經設定了預設值.

Example:

#! /usr/bin/env pythonimport collectionsdef default_factory():    return "defualt value"#在dict存在一個d[‘foo‘] = ‘bar‘ 的值。若擷取不存在於dict中的key，#則返回default_factory 函數的傳回值d = collections.defaultdict(default_factory,foo=‘bar‘) print ‘d:‘,dprint "foo=>",d.get(‘foo‘)print "foo=>",d[‘foo‘]print "var=>",d[‘bar‘]print ‘*‘ * 20# 定義的這個dict,若key 不存在於dict 中，則將這個key放到dict中，# 並對這個key賦值為listd = collections.defaultdict(list) print ‘d:‘,dprint "foo=>",d.get(‘foo‘)print "foo=>",d[‘foo‘]print "var=>",d[‘bar‘]

執行結果如下:

d: defaultdict(<function default_factory at 0x7f230849e8c0>, {‘foo‘: ‘bar‘})foo=> barfoo=> barvar=> defualt value********************d: defaultdict(<type ‘list‘>, {})foo=> Nonefoo=> []var=> []

3、class Counter

我感覺Counter 這個類比較牛，它接受一下幾種類型的參數，並返回一個dict

>>> c = Counter()                           # a new, empty counter
>>> c = Counter(‘gallahad‘)                 # a new counter from an iterable
>>> c = Counter({‘a‘: 4, ‘b‘: 2})           # a new counter from a mapping
>>> c = Counter(a=4, b=2)                   # a new counter from keyword args

看看都時什麼樣的結果：

In [1]: import collectionsIn [2]: c = collections.Counter()In [3]: print cCounter()In [4]: c = collections.Counter(‘abcdefg‘)In [5]: print cCounter({‘a‘: 1, ‘c‘: 1, ‘b‘: 1, ‘e‘: 1, ‘d‘: 1, ‘g‘: 1, ‘f‘: 1})In [6]: c = collections.Counter({‘a‘: 4, ‘b‘: 2})In [7]: print cCounter({‘a‘: 4, ‘b‘: 2})In [8]: c = collections.Counter(a=4, b=2)In [9]: print cCounter({‘a‘: 4, ‘b‘: 2})

那麼這個返回的dict 到底是一個什麼樣的結構呢? 用它自己的原話說是這樣的

Elements are stored as dictionary keys and their counts are stored as dictionary values

以下通過它本身的幾個執行個體來看具體的使用:

In [10]: c = collections.Counter(‘abcdeabcdabcaba‘)In [12]: print cCounter({‘a‘: 5, ‘b‘: 4, ‘c‘: 3, ‘d‘: 2, ‘e‘: 1})In [13]: c.most_common(3)  # 顯示出個數最多的三個元素Out[13]: [(‘a‘, 5), (‘b‘, 4), (‘c‘, 3)]In [14]: sorted(c)  # 過濾排重Out[14]: [‘a‘, ‘b‘, ‘c‘, ‘d‘, ‘e‘]In [17]: d = collections.Counter(‘simsalabim‘)In [18]: c.update(d)  # 將d 這個counter 更新到c 中In [19]: print cCounter({‘a‘: 7, ‘b‘: 5, ‘c‘: 3, ‘d‘: 2, ‘i‘: 2, ‘m‘: 2, ‘s‘: 2, ‘e‘: 1, ‘l‘: 1})

總之，Counter 類 所對應的執行個體，可以參考dict 的方法去操作。在上例中可以使用help(c) 去獲得相應的協助資訊。

本文出自 “學習筆記” 部落格，請務必保留此出處http://unixman.blog.51cto.com/10163040/1651989

Python 中的collections 模組