Python Counter()計數工具

Table of Contents

• 1. class collections.Counter([iterable-or-mapping])
  • 1.1. 例子
  • 1.2. 使用實例
• 2. To Be Continued

class collections.Counter([iterable-or-mapping])

Counter 是實現的 dict 的一個子類，可以用來方便地計數。

例子

舉個計數的例子，需要統計一個文件中，每個單詞出現的次數。實現方法如下

# 普通青年
d = {}
with open('/etc/passwd') as f:
    for line in f:
        for word in line.strip().split(':'):
            if word not in d:
                d[word] = 1
            else:
                d[word] += 1

# 文藝青年
d = defaultdict(int)
   with open('/etc/passwd') as f:
       for line in f:
           for word in line.strip().split(':'):
               d[word] += 1

# 棒棒的青年
word_counts = Counter()
with open('/etc/passwd') as f:
    for line in f:
word_counts.update(line.strip().split(':'))

使用實例

可以像下面例子一樣來創建一個 Counter:

>>> c = Counter()                           # 創建一個新的空counter
>>> c = Counter('abcasdf')                  # 一個迭代對象生成的counter
>>> c = Counter({'red': 4, 'yello': 2})      # 一個映射生成的counter
>>> c = Counter(cats=2, dogs=5)             # 關鍵字參數生成的counter

# counter 生成counter, 雖然這里並沒有什么用
>>> from collections import Counter
>>> c = Counter('abcasd')
>>> c
Counter({'a': 2, 'c': 1, 'b': 1, 's': 1, 'd': 1})
>>> c2 = Counter(c)
>>> c2
Counter({'a': 2, 'c': 1, 'b': 1, 's': 1, 'd': 1})

因為 Counter 實現了字典的 __missing__ 方法， 所以當訪問不存在的key的時候，返回值為0:

>>> c = Counter(['apple', 'pear'])
>>> c['orange']
0

counter 常用的方法：

# elements() 按照counter的計數，重復返回元素
>>> c = Counter(a=4, b=2, c=0, d=-2)
>>> list(c.elements())
['a', 'a', 'a', 'a', 'b', 'b']

# most_common(n) 按照counter的計數，按照降序，返回前n項組成的list; n忽略時返回全部
>>> Counter('abracadabra').most_common(3)
[('a', 5), ('r', 2), ('b', 2)]

# subtract([iterable-or-mapping]) counter按照相應的元素，計數相減
>>> c = Counter(a=4, b=2, c=0, d=-2)
>>> d = Counter(a=1, b=2, c=3, d=4)
>>> c.subtract(d)
>>> c
Counter({'a': 3, 'b': 0, 'c': -3, 'd': -6})

# update([iterable-or-mapping]) 不同於字典的update方法，這里更新counter時，相同的key的value值相加而不是覆蓋
# 實例化 Counter 時， 實際也是調用這個方法


# Counter 間的數學集合操作
>>> c = Counter(a=3, b=1, c=5)
>>> d = Counter(a=1, b=2, d=4)
>>> c + d                       # counter相加, 相同的key的value相加
Counter({'c': 5, 'a': 4, 'd': 4, 'b': 3})
>>> c - d                       # counter相減, 相同的key的value相減，只保留正值得value
Counter({'c': 5, 'a': 2})
>>> c & d                       # 交集:  取兩者都有的key,value取小的那一個
Counter({'a': 1, 'b': 1})
>>> c | d                       # 並集:  匯聚所有的key, key相同的情況下，取大的value
Counter({'c': 5, 'd': 4, 'a': 3, 'b': 2})

常見做法:
sum(c.values())                 # 繼承自字典的.values()方法返回values的列表，再求和
c.clear()                       # 繼承自字典的.clear()方法，清空counter
list(c)                         # 返回key組成的list
set(c)                          # 返回key組成的set
dict(c)                         # 轉化成字典
c.items()                       # 轉化成(元素，計數值)組成的列表
Counter(dict(list_of_pairs))    # 從(元素，計數值)組成的列表轉化成Counter
c.most_common()[:-n-1:-1]       # 最小n個計數的(元素，計數值)組成的列表
c += Counter()                  # 利用counter的相加來去除負值和0的值

Next

下一篇將從源碼層次剖析collection.Counter的實現, 敬請期待啦！
參見這兒: http://www.cnblogs.com/nisen/p/6055980.html