python - dictionary of dictionary - 如果键不存在，如何更新或创建值？

Question

我有以下代码要添加或更新(如果已经存在)dict()-dict 中的值:

if id not in self.steps:
    self.steps[ id ] = step
else:
    self.steps[ id ].update(step)

这工作正常，但我想知道我是否可以简化代码。我尝试使用 get，但如果键不存在，get 似乎返回元素的副本(或者只是一个新元素，与丢失的键无关？)。

self.steps.get( id ).update(step)

我有哪些选择？

Answer 1

您可以使用 collections.defaultdict ，内部值为字典。示例 -

>>> from collections import defaultdict
>>> d = defaultdict(dict)
>>> d1 = {1:2,3:4}
>>> d['id1'].update(d1)
>>> d
defaultdict(<class 'dict'>, {'id1': {1: 2, 3: 4}})