如何使用 try/except 块优化代码？

Question

我有 function 之类的：

def get_nested_dicts(items: dict, *keys, default=None):
    for key in keys:
        try:
            items = items[key]
        except KeyError:
            return default
        else:
            if not isinstance(items, dict):
                break

    return items

主要思想是优化嵌套字典的工作，例如，我有如下的收缩：

字典 = {'a': {'b': {'c': 10}}}

Output 将类似于：

print(get_nested_dicts(items, 'a', 'b', 'c')) # 10 
print(get_nested_dicts(items, 'a', 'c', 'b')) # None
print(get_nested_dicts(items, 'd', 'e', 'g')) # None

并且想要更多地优化它，以更pythonic的方式制作。 是否有一些解决方案可以优化它？

Answer 1

您可以使用dict.get()如下：

def get_nested_dicts(items: dict, *keys, default=None):
    for key in keys:
        items = items.get(key, default)
        if not isinstance(items, dict):
            break

    return items

输入

dictionary = {'a': {'b': {'c': 10}}}

print(get_nested_dicts(dictionary, 'a', 'b', 'c')) 
print(get_nested_dicts(dictionary, 'a', 'c', 'b')) 
print(get_nested_dicts(dictionary, 'd', 'e', 'g')) 

Output

10
None
None

这里get()需要两个 arguments 键和默认值返回。 如果键存在于字典中，则返回与键对应的值，否则返回None 。

Answer 2

我想， .get() function 会帮助你稍微优化一下。

def get_nested_dicts(items: dict, *keys, default=None):
    for key in keys:
        items = items.get(key, default)
        if not isinstance(items, dict): break
    return items

另一种 Pythonic 方法是使用通配符 - 'reduce'。

from functools import reduce

def get_nested_dicts(items: dict, *keys, default=None):
    return reduce(lambda v, k: v.get(k, default) if v else default, keys, items)
    

谢谢。