您好,登錄后才能下訂單哦!
我們經常遇到各種字典套字典的數據,例如:
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 }
你肯定想到了使用遞歸來解決這個問題,那么你可以試一試,看看你的遞歸函數有多少行代碼。
今天,我們使用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關鍵字來解決字典與列表混合嵌套的情況。
總結
以上就是這篇文章的全部內容了,希望本文的內容對大家的學習或者工作具有一定的參考學習價值,謝謝大家對億速云的支持。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。