也许你低估了defaultdict的偷懒能力！

今天来和大家聊聊日常经常使用到的偷懒方法 --> defaultdict

defaultdict示例

字典作为日常使用频率较高的一种数据类型，常会遇到判断key是否在字典中的情况。

这时，我们是否按照如下代码写的：

d = dict()
if 'key' in d:
    d['key'] += 1
else:
    d['key'] = 0
print(d['key']) # 1

我们每次都需要判断后再进行相关操作。

但是，当我们使用了defaultdict后，就可以偷懒的简化if else 的格式了：

from collections import defaultdict

d = defaultdict(int)
d["key"] += 1
print(d['key']) # 1

关于defaultdict

大家日常使用到defaultdict的场景，绝大多数都是上述举例为了减少if else的判断。

当然除了默认的int初始化，还有列表追加 d = defaultdict(list) 的无脑append操作。

可是，我们是否有深挖过defaultdict的其他场景呢？来先看看它的源码：

    def __init__(self, default_factory=None, **kwargs): 
        # known case of _collections.defaultdict.__init__
        """
        defaultdict(default_factory=None, /, [...]) --> 
        dict with default factory
        
        The default factory is called without arguments to produce
        a new value when a key is not present, in __getitem__ only.
        A defaultdict compares equal to a dict with the same items.
        All remaining arguments are treated the same as if they were
        passed to the dict constructor, including keyword arguments.
        
        # (copied from class doc)
        """
        pass

源码注释中，只是简单说明了使用default_factory参数，可以让调用键不存在时生成新值。

就比如我们上面 defaultdict(int) 将key不存在时，value默认赋值0， d = defaultdict(list) 将key不存在时，value默认赋值空列表。

听起来功能就是如此了吧...

现在我们来换个场景，拿同学们打力扣周赛举个例子。

力扣周赛

每个人参加力扣周赛时，会根据我们的比赛结果进行评分。如果是第一次参加比赛，因为没有初始积分，从0分开始不太合适。

所以每位选手的初始基准分为1500分。这样就可以根据选手分数来考量本次比赛表现进行加分了。

先来看看默认字典的代码应该如何操作：

scores = dict()
add_score = 10
# 方法1
if 'xiaoming' in scores:
    scores['xiaoming'] += add_score
else:
    scores['xiaoming'] = 1500 + add_score

# 方法2
scores['xiaoming'] = scores.get('xiaoming', 1500) + add_score

现在我们想使用defaultdict，但defaultdict如果赋值int，就没办法提供这个基准分了，该如何是好？

此时我们应该深入理解下 default_factory ，它不仅仅支持我们传入默认的int、list，还支持我们使用自定义函数。

from collections import defaultdict

def diy_func():
    print("init user score.")
    return 1500

scores = defaultdict(diy_func)
scores['xiaoming'] += 10
print(scores['xiaoming'])
# init user score.
# 1510

我们通过自定义一个函数赋值给 default_factory， 帮我们初始化用户的分数。

但这个自定义的函数方法有些太过单一了，是否可以简化？此时我们应该考虑到lambda表达式啊！

scores = defaultdict(lambda: 1500)
scores['xiaoming'] += 10
print(scores['xiaoming'])

这样做是不是就更简洁方便了。有没有觉得这操作很nice？

然而，我们注意到defaultdict的 __init__ 方法是存在 **kwargs** 参数，我们还没有使用呢！

它还可以这么玩：

from collections import defaultdict

data = defaultdict(xiaozhang=1600, xiaowang=1700)

scores = defaultdict(lambda: 1500, data)
scores['xiaoming'] += 10
scores['xiaowang'] -= 15
print(scores.items())
# dict_items([('xiaozhang', 1600), ('xiaowang', 1685), ('xiaoming', 1510)])

所以，defaultdcit可以在 使用 default_factory 的基础上，导入初始的字典进行。是不更溜了？

关于default_dict的内容，今天就学到这里吧，希望对大家有所帮助。

欢迎关注我的公_众号: 清风Python，带你每日学习Python算法刷题的同时，了解更多python小知识。

我的个人博客：https://qingfengpython.cn

力扣解题合集：https://github.com/BreezePython/AlgorithmMarkdown