Python步步飞升之collections.Counter
1. 前言
Python内置四种基本container：list, dict, set, tuple，collections模块为其补充。Counter为collections中一个容器。一言以蔽之，是为计数器.
2. 从一个简单例子说起
有一个list
l = ['red', 'blue', 'red', 'green', 'blue', 'blue']
现在需要统计下各个颜色出现的次数，用一个循环实现
from collections import defaultdict
l = ['red', 'blue', 'red', 'green', 'blue', 'blue']
counter = defaultdict(int)
for i in l:
counter[i] += 1
print(counter)
运行结果
defaultdict(<class 'int'>, {'red': 2, 'blue': 3, 'green': 1})
这里用到了以前讲述到的defaultdict。
3. Life is short, you need Python
现在有了collections.Counter工具，我们可以更加便利的实现这点
from collections import Counter
l = ['red', 'blue', 'red', 'green', 'blue', 'blue']
counter = Counter(l)
print(counter)
运行结果
Counter({'blue': 3, 'red': 2, 'green': 1})
4. Counter简单介绍
class collections.Counter([iterable-or-mapping])
Counter是dict的一个子类，构造函数接受iterable或者mapping，用以计数。
例举创建Counter的方法
c = Counter() # 空Counter
c = Counter('gallahad') # 从iterable创建
c = Counter({'red': 4, 'blue': 2}) # 从mapping创建
c = Counter(cats=4, dogs=8) # 以关键字参数创建
5. Mark一些该类常用方法和特点
5.1 Counter不会遇到“KeyError”异常
>>> d = dict()
>>> d['a']
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
KeyError: 'a'
>>> c = Counter()
>>> print(c['a'])
0
5.2 Counter值可为零和负数
>>> c = Counter({'a': 1, 'b': -1, 'c': 0})
>>> c['b']
-1
>>> c['c']
0
5.3 Counter支持“加减或与”运算
>>> c = Counter(a=3, b=1)
>>> d = Counter(a=1, b=2)
>>> c + d # 两个counter对应值相加: c[x] + d[x]
Counter({'a': 4, 'b': 3})
>>> c - d # 两个counter对应值相减 (只保留正数值)
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})
其中一元加减操作表示被加减一个空Counter
>>> c = Counter(a=2, b=-4)
>>> +c
Counter({'a': 2})
>>> c = Counter(a=2, b=-4)
>>> -c
Counter({'b': 4})
所以刚好可以使用 +c 用来去除Counter中非正数值
5.4 elements()
以任意顺序返回Counter中key，个数为value值的迭代器。
>>> c = Counter('askdlsajpdsadlsa')
>>> print(c)
Counter({'a': 4, 's': 4, 'd': 3, 'l': 2, 'k': 1, 'j': 1, 'p': 1})
>>> list(c.elements())
['a', 'a', 'a', 'a', 's', 's', 's', 's', 'k', 'd', 'd', 'd', 'l', 'l', 'j', 'p']
>>> list(sorted(c.elements()))
['a', 'a', 'a', 'a', 'd', 'd', 'd', 'j', 'k', 'l', 'l', 'p', 's', 's', 's', 's']
5.5 most_common([n])
返回次数最高的N个mapping组成的list
>>> Counter('askdlsajpdsadlsa').most_common(3)
[('a', 4), ('s', 4), ('d', 3)]
当n未输入时返回所有，按照次数高低排序
>>> Counter('askdlsajpdsadlsa').most_common()
[('a', 4), ('s', 4), ('d', 3), ('l', 2), ('k', 1), ('j', 1), ('p', 1)]
那么可以简单的获得次数最低的的3个mapping组成的list
>>> Counter('askdlsajpdsadlsa').most_common()[:-4:-1]
[('p', 1), ('j', 1), ('k', 1)]
6. 结语
如有疑问，欢迎留言共同探讨。
