python - 在嵌套字典中搜索关键树

Question

我想检查嵌套字典中是否存在键元组，类似于dict.get。其功能可以通过如下方式实现。

nested_dict = {
    'x': 1,
    'a': {
        'b': {
            'c': 2,
            'y': 3
        },
        'z': 4
    }
}

def check(nested, *path):
    for key in path:
        if isinstance(nested, dict) and key in nested:
            nested = nested[key]
        else:
            return False
    return True


check(nested_dict, 'a', 'b', 'c')  # True
check(nested_dict, 'x')  # True
check(nested_dict, 'a', 'b', 'y')  # True
check(nested_dict, 'a', 'z')  # True

check(nested_dict, 'y')  # False
check(nested_dict, 'a', 'y')  # False
check(nested_dict, 'a', 'b', 'c', 'y')  # False

是否有更清晰的(或者更好的是内置的)方法来执行此操作？

Answer 1

对于 python 3.x，请执行 from functools import reduce。

您可以包装 try .. except KeyError (和 TypeError)并返回适当的 bool 值:

>>> reduce(lambda x,y: x[y], ["a", "b", "c"], nested_dict)
2
>>> reduce(lambda x,y: x[y], ["a", "b", "d"], nested_dict)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <lambda>
KeyError: 'd'

<小时/>

PS:这些行有时写起来很有趣。但老实说，我会在任何生产代码中使用您的版本。