Python-collections模块之defaultdict

defaultdict

defaultdict 是 dict 类型的子类,正如其名,初始化时,可以给key指定默认值,什么意思呢?直接看代码。
如果是普通的dict对象,访问一个不存在的key时,会报错:

dict1 = dict()
print(dict1['a'])
"""
Traceback (most recent call last):
  File "D:/Projects/test/t1.py", line 14, in <module>
    print(dict1['a'])
KeyError: 'a'
"""

而使用defaultdict,可以给不存在的key给一个默认的初始值,例如:
默认空列表的dict:

from collections import defaultdict

res1 = defaultdict(list)
print(res1['a'])  # []
print(res1.get('a'))  # []
print(res1.get('a', 'hello world 1'))  # []

关于defaultdict的一些应用场景:
场景一:我们经常想要将一个列表中的数据进行整合,比如这种:

from collections import defaultdict

initial_data = [('class1', '张三'), ('class2', '李四'), ('class1', '王五'),
        ('class3', '赵六'), ('class3', '李刚弹'), ('class2', '王大炮'),
        ('class1', '王大锤'), ('class2', '李二狗'), ('class1', '佚名')
        ]
# ==> 变成:
finally_data = {
    'class1': ['张三', '王五', '王大锤', '佚名'],
    'class2': ['李四', '王大炮', '李二狗', ],
    'class3': ['赵六', '李刚弹']
}
# 如果自己实现的话可能会用for循环依次判断,又或者使用到setdefault方法也可以
# 这里如果使用defaultdict方法算是一种比较优雅的解决吧:
res5 = defaultdict(list)
for k, v in initial_data:
    res5[k].append(v)

参考:https://www.cnblogs.com/suguangti/p/13150567.html

posted @ 2021-02-10 17:06  bokeyuan6  阅读(430)  评论(0编辑  收藏  举报