Python每日一练0007

xiaoxiao2021-02-28  17

问题

怎样实现一个键对应多个值的字典(也叫 multidict)?

例如:

d = { 'a' : [1, 2, 3], 'b' : [4, 5] } e = { 'a' : {1, 2, 3}, 'b' : {4, 5} }

解决方案

使用collections模块中的defaultdict类

defaultdict的特点是,当key不存在时,会自动创建它,也就是说我们不用再像dict一样首先判断key是否存在了,下面是一个例子:

>>> from collections import defaultdict >>> d = defaultdict(list) >>> d['a'].append(1) >>> d['a'].append(2) >>> d['a'].append(3) >>> d['b'].append(4) >>> d['b'].append(5) >>> print(d) defaultdict(<class 'list'>, {'a': [1, 2, 3], 'b': [4, 5]})

当然,也可以用dict的setdefault函数来实现同样的效果,但看起来会比较“丑陋”

>>> d = dict() >>> d.setdefault('a', []).append(1) >>> d.setdefault('a', []).append(2) >>> print(d) {'a': [1, 2]}

讨论

根据Python3官方文档对defaultdict的介绍可以了解到

defaultdict是dict的子类,defaultdict重写了dict的__missing__方法,并且增加了一个可以修改的default_factory成员

当defaultdict的key不存在时,会将key初始化为调用default_factory得到的返回值,default_factory可以在初始化时指定,也可以随后再进行改变

>>> from collections import defaultdict >>> d = defaultdict(list) >>> d['l'].append(1) >>> d.default_factory = set >>> d['s'].add(2) >>> print(d) defaultdict(<class 'set'>, {'l': [1], 's': {2}})

来源

Python Cookbook

关注

欢迎关注我的微信公众号:python每日一练

转载请注明原文地址: https://www.6miu.com/read-1450100.html

最新回复(0)