python - 动态地在字典中设置一个值

Question

如果我有一个嵌套的字典 d = {'a':{'b':{}}}和一个字符串 'a.b.c'和一个值 'X'
我需要根据键字符串将值放入字典中。

我想要实现的可以硬编码为 d['a']['b']['c'] = 'X'但我需要动态地做。 key 串可以是任意长度。

对于奖励积分:如果 key 不存在，我还需要创建 key ，例如 'a.b.z'但我相信我能弄清楚如果我能解决它们已经存在的情况。

Answer 1

def set(d, key, value):
    dd = d
    keys = key.split('.')
    latest = keys.pop()
    for k in keys:
        dd = dd.setdefault(k, {})
    dd.setdefault(latest, value)

d = {}
set(d, 'a.b.c', 'X')
set(d, 'a.b.d', 'Y')
print(d)

结果:

{'a': {'b': {'c': 'X', 'd': 'Y'}}}