python - 编辑字典值python的正确方法

Question

我用两种不同的方式编写了以下代码。我正在尝试找到“正确的 pythonic”方法。我将解释两者的原因。

第一种方式，EAFP。这个使用pythons EAFP原理，但是会导致一些代码重复。

try:
    my_dict['foo']['bar'] = some_var
except KeyError:
    my_dict['foo'] = {}
    my_dict['foo']['bar'] = some_var

第二种方式，LBYL。 LBYL 不完全被认为是 pythonic，但它删除了代码重复。

if 'foo' not in my_dict:
    my_dict['foo'] = {}

my_dict['foo']['bar'] = some_var

哪种方式最好？或者有更好的方法吗？

Answer 1

我会说经验丰富的 Python 开发人员要么使用

dict.setdefault或 collections.defaultdict

my_dict.setdefault('foo', {})['bar'] = some_var

或

from collection import defaultdict
my_dict = defaultdict(dict)
my_dict['foo']['bar'] = some_var

同样为了完整起见，我将向您介绍一个递归的defaultdict 模式，它允许无限深度的字典而不会出现任何键错误

>>> from collections import defaultdict
>>> def tree(): return defaultdict(tree)

>>> my_dict = tree()
>>> my_dict['foo']['bar']['spam']['egg'] = 0
>>> my_dict
defaultdict(<function tree at 0x026FFDB0>, {'foo': defaultdict(<function tree at 0x026FFDB0>, {'bar': defaultdict(<function tree at 0x026FFDB0>, {'spam': defaultdict(<function tree at 0x026FFDB0>, {'egg': 0})})})})