python - 从嵌套字典中获取值

Question

是否有从嵌套字典中获取值的标准方法？该函数相对容易编写，但我很好奇 PSL 或语言本身是否已有可以以这种方式使用的东西？

以下是我的意思的示例:

def grab_from_dict(d, *keys):
    assert isinstance(d, dict), 'd must be of type dict'

    current_dict = d
    for key in keys[0 : -1]:
        if key not in current_dict:
            return None

        current_dict = current_dict[key]

    if current_dict is None:
        return None

    return current_dict.get(keys[-1], None)

d = {
    'a' : {
        'b' : {
           'c1' : {
                'd' : 'leeloo'
            },
           'c2' : {
                'd' : None
            },
           'c3' : {
                'e' : None
            },
            'c4' : None
        }
    }
}

print grab_from_dict(d, 'a', 'b', 'c1')
> {'d': 'leeloo'}
print grab_from_dict(d, 'a', 'b', 'c1', 'd')
> leeloo
print grab_from_dict(d, 'a', 'b', 'c2')
> {'d': None}
print grab_from_dict(d, 'a', 'b', 'c2', 'd')
> None
print grab_from_dict(d, 'a', 'b', 'c3')
> {'e': None}
print grab_from_dict(d, 'a', 'b', 'c3', 'd')
> None
print grab_from_dict(d, 'a', 'b', 'c4')
> None
print grab_from_dict(d, 'a', 'b', 'c4', 'd')
> None
print grab_from_dict(d, 'a', 'b', 'c5')
> None
print grab_from_dict(d, 'a', 'b', 'c5', 'd')
> None

这给了我一种方法来获取嵌套字典深处的值，而不必担心父字典的存在。因此，不要这样写:

value = None
if 'a' in d and d['a'] not None:
    if 'b' in d['a'] and d['a']['b'] is not None:
        if 'c1' in d['a']['b'] and d['a']['b']['c1'] is not None:
            value = d['a']['b']['c1'].get('d', None)
print value
> leeloo

我可以这样写:

value = grab_from_dict(d, 'a', 'b', 'c1', 'd')
print value
> leeloo

如果缺少任何父级，该函数将简单地返回 None:

value = grab_from_dict(d, 'a', 'a', 'c1', 'd')
print value
> None

Answer 1

捕获异常应该有效:

try:
    result = d['a']['b']['c1']
except KeyError:
    result = None