counter的用法python|极客教程

counter的用法python

在Python中，Counter是collections模块中的一个数据结构，它用来对可迭代对象中的元素进行计数。Counter是一个字典的子类，在内部使用字典来存储元素和它们对应的计数。Counter对象可以用于统计一组数据中各个元素的出现次数，在数据分析和处理中经常会用到。

创建Counter对象

我们可以使用Counter()函数来创建一个Counter对象，并传入一个可迭代的对象作为参数，比如一个列表、元组或字符串。下面是创建Counter对象的几种方式：

from collections import Counter

# 从列表创建Counter对象
lst = ['a', 'b', 'c', 'a', 'b', 'a']
counter_lst = Counter(lst)
print(counter_lst)  # Counter({'a': 3, 'b': 2, 'c': 1})

# 从元组创建Counter对象
tpl = ('a', 'b', 'c', 'a', 'b', 'a')
counter_tpl = Counter(tpl)
print(counter_tpl)  # Counter({'a': 3, 'b': 2, 'c': 1})

# 从字符串创建Counter对象
s = 'abcaab'
counter_str = Counter(s)
print(counter_str)  # Counter({'a': 3, 'b': 2, 'c': 1})

访问Counter中的元素

Counter对象可以像字典一样通过键来访问对应的计数值。我们可以使用键来查询某个元素的出现次数，也可以使用items()方法来遍历Counter对象中的所有元素及其计数值。

from collections import Counter

lst = ['a', 'b', 'c', 'a', 'b', 'a']
counter = Counter(lst)

# 查询某个元素的计数值
print(counter['a'])  # 3

# 遍历Counter对象中的所有元素及其计数值
for key, value in counter.items():
    print(key, value)

更新Counter对象

Counter对象支持与其他Counter对象相加、相减、取交集等操作，这些操作会更新Counter对象的计数值。

from collections import Counter

lst1 = ['a', 'b', 'c', 'a', 'b', 'a']
counter1 = Counter(lst1)

lst2 = ['a', 'b', 'd', 'a']
counter2 = Counter(lst2)

# 相加，计数值相加
counter_sum = counter1 + counter2
print(counter_sum)  # Counter({'a': 5, 'b': 3, 'c': 1, 'd': 1})

# 相减，计数值相减（结果小于等于0的计数值会被丢弃）
counter_diff = counter1 - counter2
print(counter_diff)  # Counter({'a': 2, 'b': 1, 'c': 1})

# 取交集，计数值取较小的值
counter_intersection = counter1 & counter2
print(counter_intersection)  # Counter({'a': 2, 'b': 1})

其他常用方法

除了上述的操作，Counter对象还有一些其他常用的方法：

elements()：返回一个迭代器，包含所有元素，重复元素多次出现
most_common(n)：返回出现次数最多的n个元素及其计数值
subtract()：扣除传入的元素计数值

from collections import Counter

lst = ['a', 'b', 'c', 'a', 'b', 'a']
counter = Counter(lst)

# 返回所有元素
print(list(counter.elements()))  # ['a', 'a', 'a', 'b', 'b', 'c']

# 返回出现次数最多的两个元素
print(counter.most_common(2))  # [('a', 3), ('b', 2)]

# 扣除计数值
counter.subtract({'a': 2, 'b': 1, 'd': 1})
print(counter)  # Counter({'a': 1, 'b': 1, 'c': 1, 'd': -1})