如何在 Python 中展平嵌套字典并压缩其键？

通过键压缩展平嵌套字典

在代码中处理复杂的数据结构时，“展平”它们通常很有用以便于操作。对于嵌套字典来说尤其如此，其中多个级别的键和值会增加操作的复杂性。在本文中，我们将探讨如何在压缩键的同时展平嵌套字典。

问题

考虑以下嵌套字典：

{'a': 1,
 'c': {'a': 2,
       'b': {'x': 5,
             'y': 10}},
 'd': [1, 2, 3]}

目标是将这本字典扁平化为某种东西例如：

{'a': 1,
 'c_a': 2,
 'c_b_x': 5,
 'c_b_y': 10,
 'd': [1, 2, 3]}

其中嵌套键被压缩以创建单个扁平键。

解决方案

扁平化嵌套字典，我们可以使用 collections.abc 模块中的 flatten() 函数。该函数可以定义如下：

def flatten(dictionary, parent_key='', separator='_'):
    items = []
    for key, value in dictionary.items():
        new_key = parent_key + separator + key if parent_key else key
        if isinstance(value, MutableMapping):
            items.extend(flatten(value, new_key, separator=separator).items())
        else:
            items.append((new_key, value))
    return dict(items)

使用函数

要使用 flatten() 函数，只需传入嵌套字典作为参数即可。该函数将递归地遍历字典，“展平”它并压缩键。

>>> flatten({'a': 1, 'c': {'a': 2, 'b': {'x': 5, 'y': 10}}, 'd': [1, 2, 3]})
{'a': 1, 'c_a': 2, 'c_b_x': 5, 'd': [1, 2, 3], 'c_b_y': 10}

如您所见，展平的字典具有单层键，键被压缩以合并嵌套原始词典的结构。

以上是如何在 Python 中展平嵌套字典并压缩其键？的详细内容。更多信息请关注PHP中文网其他相关文章！