如何在 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中文網其他相關文章！