使用yield壓平嵌套字典的方法

小編給大家分享一下使用yield壓平嵌套字典的方法，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

我們經常遇到各種字典套字典的數據，例如：

nest_dict = {
    'a': 1,
    'b': {
        'c': 2,
        'd': 3,
        'e': {'f': 4}
    },
    'g': {'h': 5},
    'i': 6,
    'j': {'k': {'l': {'m': 8}}}
}

有沒有什么簡單的辦法，把它壓扁，變成：

{
    'a': 1,
    'b_c': 2,
    'b_d': 3,
    'b_e_f': 4,
    'g_h': 5,
    'i': 6,
    'j_k_l_m': 8
}

推薦學習《Python視頻教程》

你肯定想到了使用遞歸來解決這個問題，那么你可以試一試，看看你的遞歸函數有多少行代碼。

今天，我們使用yield關鍵字來實現這個需求，在不炫技的情況下，只需要8行代碼。在炫技的情況下，只需要3行代碼。

要快速地把這個嵌套字典壓扁，我們需要從下網上來處理字段。例如對于b->e->f->4這條路徑，我們首先把最里面的{'f': 4}轉換為一個元組('f', 4)。然后，把這個元組向上拋出，于是得到了元組('e', ('f', 4))。我們把 e拼接到f的前面，變為：('e_f', 4)，繼續往上拋出，得到('b', ('e_f', 4))。再把b拼接到e_f上面，得到('b_e_f', 4)。完成一條線路的組裝。

這個邏輯如果使用yield關鍵字來實現，就是：

def flat(x):
    for key, value in x.items():
        if isinstance(value, dict):
            for k, v in flat(value):
                k = f'{key}_{k}'
                yield (k, v)
        else:
            yield (key, value)

運行結果如下圖

通過使用 yield關鍵字，字典的key會像是在流水線上一樣，一層一層從內向外進行組裝，從而形成完整的路徑。

以上是使用yield壓平嵌套字典的方法的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！