💡 如果想阅读最新的文章,或者有技术问题需要交流和沟通,可搜关注微信公众号号号“希望睿智”。
概述在上一节,我们介绍了Python的sqlite3模块,包括:sqlite3模块中一些常用的函数和类。在这一节,我们将介绍Python的collections模块。collections模块是Python中的内置模块,它实现了特殊的容器数据类型,提供了Python的通用内建容器的替代选择。
使用collections模块中的这些数据结构,可以帮助我们更有效地解决问题,并写出更简洁、更易于理解的代码。比如:如果我们需要一个字典,当尝试获取一个不存在的键时,返回一个默认值,而不是抛出一个异常,那么可以使用defaultdict。又比如:如果我们需要一个记住元素插入顺序的字典,那么可以使用OrderedDict。
下面,我们将逐一介绍collections模块中一些常用的函数和类。
collections.Counter类collections.Counter类是一个字典子类,用于计数可哈希的对象。它返回一个字典,对应的键是输入值,值是输入值出现的次数。这是一个非常有用的工具,可以快速计数元素的数量。
from collections import Counterresult = Counter('hello')# 输出:Counter({'l': 2, 'h': 1, 'e': 1, 'o': 1})print(result)
在上面的示例代码中,向Counter类的构造函数传入了字符串。返回的result为一个字典,包含了每个字符及其出现的次数。当然,Counter也可以接受列表、元组、集合或任何其他可迭代对象作为参数。
collections.deque类collections.deque类是一个双端队列,允许从队列的两端进行添加和弹出操作。双端队列是一种具有两个端点的队列,可以在队列的两端添加和弹出元素。deque支持线程安全、内存高效地添加(append)和弹出(pop),从两端都可以操作,具有GIL锁的线程安全。
from collections import deque# 创建一个空的dequed = deque()# 在deque的右侧添加元素d.append('h')d.append('e')d.append('l')d.append('l')d.append('o')# 在deque的左侧添加元素d.appendleft('C')d.appendleft('S')d.appendleft('D')d.appendleft('N')# 弹出deque右侧的元素element = d.pop()# 输出:'o'print(element)# 弹出deque左侧的元素element = d.popleft()# 输出:'N' print(element)
collections.defaultdict类collections.defaultdict类是一个字典子类,用于创建一个类似于字典的数据结构。它覆盖了字典的getitem方法,以便当请求的键不存在时,可以返回一个默认值,而不是抛出KeyError异常。默认值是在创建defaultdict对象时通过参数指定的,如果默认值是一个工厂函数,则该函数将被调用以生成默认值。
from collections import defaultdict# 创建一个defaultdict,默认值为0dd = defaultdict(int)# 添加几个键值对dd['C'] = 66dd['S'] = 99# 访问一个不存在的键,返回默认值0print(dd['D'])
collections.namedtuple函数collections.namedtuple函数用于创建一个可以使用名字来访问元素内容的tuple子类。namedtuple非常有用,特别是当我们需要创建一些具有多个字段的数据结构时。相比使用字典或列表,namedtuple更加轻量级,并且可以直接通过属性名访问元素,而不需要使用索引。这就避免了在代码中使用数字索引来访问元组或列表的元素,提高了代码的可读性和可维护性。
from collections import namedtuple# 创建一个namedtuple类,名为Point,具有xpos和ypos两个字段Point = namedtuple('Point', ['xpos', 'ypos'])# 创建一个Point对象point = Point(66, 99)# 访问对象的字段# 输出:66print(point.xpos)# 输出:99print(point.ypos)
collections.OrderedDict类collections.OrderedDict类用于创建一个有序的字典对象。它继承自dict类,并添加了一些额外的功能,以保持字典中元素的插入顺序。OrderedDict与普通的dict类似,但它会记住元素被插入的顺序。当遍历OrderedDict时,元素将按照它们被插入的顺序返回。使用OrderedDict可以确保字典中的元素按照一定的顺序进行遍历,而不需要额外的排序操作。
from collections import OrderedDict# 创建OrderedDict对象od = OrderedDict()# 添加元素od['C'] = 66od['S'] = 77od['D'] = 88od['N'] = 99# 遍历OrderedDictfor key, value in od.items(): print(key, value)
collections.UserList类collections.UserList类是list的子类,允许用户自定义列表的行为。UserList提供了与list相同的接口,但允许用户覆盖或添加方法以修改其行为。UserList非常有用,特别是当你需要定制列表的行为,或需要添加额外的功能时。你可以通过继承UserList类,并重写其方法来实现自定义的列表功能。
from collections import UserListclass MyList(UserList): def __init__(self, list_init = None): super().__init__(list_init) def append(self, value): print(f"item {value} inserted") super().append(value) # 创建一个MyList对象my_list = MyList([66, 77, 88])# 在列表中添加元素,输出:item 100 inserted my_list.append(100) # 访问列表中的元素,输出:66print(my_list[0])
使用UserList可以轻松地扩展列表的功能,并根据需要自定义其行为,这使得UserList成为处理特殊需求的列表操作的强大工具。除了自定义列表UserList,Python还内置了UserDict、UserSet、UserString,以允许用户定义他们自己的字典类、集合类和字符串类,这里就不再赘述了。