透過鍵壓縮展平巢狀字典
在程式碼中處理複雜的資料結構時,「展平」它們通常很有用以便於操作。對於嵌套字典來說尤其如此,其中多個層級的鍵和值會增加操作的複雜性。在本文中,我們將探討如何在壓縮鍵的同時展平嵌套字典。
問題
考慮以下巢狀字典:
{'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中文網其他相關文章!