跳转至

Python的collections包

collections 模块是 Python 标准库中提供了额外的数据结构以及一些有用的工具类的模块。这些数据结构和工具类在某些情况下比内置的数据结构(如列表、字典、集合等)更加高效和方便,可以用来解决一些常见的编程问题。以下是 collections 模块中一些常用的数据结构和类:

1. namedtuple

namedtuple 是一个工厂函数,用来创建带有命名字段的元组子类。它可以像普通的元组一样使用,但是可以通过字段名访问元素,比直接使用索引更具有可读性。

from collections import namedtuple

# 定义一个 Point 类型的 namedtuple
Point = namedtuple('Point', ['x', 'y'])

# 创建一个 Point 对象
p = Point(10, 20)

# 访问字段
print(f"x 坐标: {p.x}, y 坐标: {p.y}")

2. deque

deque 是双端队列,支持从两端快速添加和删除元素。它比列表在插入和删除操作时更高效。

from collections import deque

# 创建一个空的 deque
d = deque()

# 在右端添加元素
d.append(1)
d.append(2)

# 在左端添加元素
d.appendleft(0)

# 弹出右端元素
print(d.pop())

# 弹出左端元素
print(d.popleft())

3. Counter

Counter 是用来计数可哈希对象的计数器,它返回一个字典,其中键是元素,值是计数。常用于统计词频等场景。

from collections import Counter

# 创建一个 Counter 对象
cnt = Counter(['apple', 'banana', 'apple', 'cherry', 'banana'])

# 访问计数
print(cnt['apple'])  # 输出 2,apple 出现了两次
print(cnt['cherry'])  # 输出 1,cherry 出现了一次

# 更新计数
cnt.update(['apple', 'cherry'])

# 获取最常见的元素
print(cnt.most_common(1))  # 输出 [('apple', 3)]

4. defaultdict

defaultdict 是字典的一个子类,它接受一个工厂函数作为参数,用来为字典查询不到的键提供默认值。

from collections import defaultdict

# 创建一个 defaultdict,默认值为 int 类型的 0
d = defaultdict(int)

# 访问不存在的键,会返回默认值 0
print(d['apple'])  # 输出 0

# 设置默认值为列表,并添加元素
d = defaultdict(list)
d['fruit'].append('apple')
d['fruit'].append('banana')

# 访问键 'fruit' 的值
print(d['fruit'])  # 输出 ['apple', 'banana']

5. OrderedDict

OrderedDict 是有序字典,它记住了元素插入的顺序,可以按照插入顺序迭代。

from collections import OrderedDict

# 创建一个 OrderedDict 对象
d = OrderedDict()

# 添加键值对
d['one'] = 1
d['two'] = 2
d['three'] = 3

# 按照插入顺序迭代
for key, value in d.items():
    print(key, value)

6. 其他常见功能

除了上述几个常用的数据结构外,collections 模块还包括其他一些有用的功能,如 ChainMapUserDictUserList 等,用于处理字典链、自定义字典和列表等。这些工具类和数据结构使得在特定场景下能够更高效地进行数据处理和操作。

7. 总结

collections 模块提供了多种实用的数据结构和工具类,能够帮助开发人员在处理复杂数据和解决特定问题时提供更加高效和方便的解决方案。熟练掌握 collections 模块中的各种类和函数,能够显著提升 Python 编程的效率和代码的可读性。

评论