Python 3字典理解与子字典解包？

Question

假设有一个字典，root由key：value对组成，其中一些值本身就是字典。

一个（如果是这样，如何）可以通过字典理解解压缩这些子词典吗？

EG

{k: v if type(v) is not dict else **v for k, v in root.items()}

例：

root = {'a': 1, 'b': {'c': 2, 'd': 3}}


result = {'a': 1, 'c': 2, 'd': 3}

Answer 1

我想我应该发布一个更广泛的解释作为答案来帮助你，因为它与其他现有问题有点不同

{
    _k: _v
    for k, v in root.items()
    for _k, _v in (         # here I create a dummy dictionary if non exists
                      v if isinstance(v, dict) else {k: v}
                  ).items() # and iterate that
}

理解的关键部分是你需要一致和通用的逻辑来理解工作。

您可以通过使用v if isinstance(v, dict) else {k: v}创建以前不存在的虚拟嵌套字典来完成此操作

然后这是一个简单的嵌套字典解包练习。

为了帮助您将来理解，我建议您编写代码，例如：

res = dict()
for k,v in root.items():
    d = v if isinstance(v, dict) else {k: v}
    for _k, _v in d.items():
        res[_k] = _v

并从此向后工作

Useful references

• Putting a simple if-then-else statement on one line
• Nested dictionary comprehension python

Answer 2

如果你有几个级别的嵌套字典，我建议你基于递归函数的以下解决方案：

def flatten(res, root):
    for k,v in root.items():
        if isinstance(v, dict):
            flatten(res, v)
        else:
            res[k] = v

root = {'a': 1, 'b': {'c': 2, 'd': {'e': 5, 'f': 6}}}

result = {}
flatten(result, root)

print(result)  # {'a': 1, 'c': 2, 'e': 5, 'f': 6}

Answer 3

这是一个递归解决方案。在函数_flatten_into_kv_pairs中，如果值不是字典，我们遍历键和值对并产生这些键/值。如果是，那么我们用_flatten_into_kv_pairs结构递归调用yield from。函数flatten_dict只是一个shell，它将键/值对的序列转换回字典。

def _flatten_into_kv_pairs(dict_object):
    for k, v in dict_object.items():
        if isinstance(v, dict):
            yield from _flatten_into_kv_pairs(v)
        else:
            yield k, v


def flatten_dict(dict_object):
    return dict(_flatten_into_kv_pairs(dict_object))


root = {'a': 1, 'b': {'c': 2, 'd': 3, 'e': {'f': 4, 'g': 5}}}
print(flatten_dict(root))

输出：

{'a': 1, 'c': 2, 'd': 3, 'f': 4, 'g': 5}