使用 yield 壓平嵌套字典有多簡單？

• 2019 年 11 月 7 日
• 筆記

攝影：產品經理

買單：kingname

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

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關鍵字來解決字典與列表混合嵌套的情況。

推薦閱讀：一日一技：如何把多層嵌套的列表展平