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

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

總結

以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對我們的支援。