使用 yield 壓平多層嵌套字典列表混合數據

• 2019 年 11 月 7 日
• 筆記

攝影：產品經理

買單：kingname

在上一篇文章裡面，我們講到了如何使用Python的yield關鍵字簡化程式碼，壓平多層嵌套字典的。

那麼如果我們的數據不僅僅有字典，還有列表，是一個字典列表多層嵌套的數據怎麼辦呢？例如：

nest_dict = {      'a': 1,      'b': {          'c': 2,          'd': 3,          'e': {'f': 4}      },      'g': {'h': 5},      'i': 6,      'j': {'k': {'l': {'m': 8}}},      'n': [1, {'o': 1, 'p': [1, 2, 3],  'q': {'r': {'s': 100}}}, 3, [1, 2, 3], 5]  }

現在，請停下來，敲一敲程式碼，想想如何把處理列表的邏輯添加進去。

首先，我們來看一下最終被壓平以後的數據長什麼樣：

{'a': 1,   'b_c': 2,   'b_d': 3,   'b_e_f': 4,   'g_h': 5,   'i': 6,   'j_k_l_m': 8,   'n_0': 1,   'n_1_o': 1,   'n_1_p_0': 1,   'n_1_p_1': 2,   'n_1_p_2': 3,   'n_1_q_r_s': 100,   'n_2': 3,   'n_3_0': 1,   'n_3_1': 2,   'n_3_2': 3,   'n_4': 5}

對於'n': ['a', 'b', 'c']這種形式的數據，我們把它轉換為：{'n_0': 'a', 'n_1': 'b', 'n_2': 'c'}

我們原來的核心程式碼是這樣的：

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)

你的第一反應，是不是這樣修改程式碼：

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)          elif isinstance(value, list):              "一大堆處理列表的程式碼"          else:              yield (key, value)

如果你使用return和遞歸，你可能確實需要這樣寫。

但如果你使用yield關鍵字，那麼，你雖然也要修改程式碼，可是修改的地方卻不是這裡。我們要修改的地方在for key, value in x.items()。

因為.items()這個方法是字典的方法，列表沒有這個方法。所以我們需要寫一個通用的迭代生成器，支援字典和列表，所以我們增加一個函數：iter_x:

def iter_x(x):      if isinstance(x, dict):          for key, value in x.items():              yield (key, value)      elif isinstance(x, list):          for index, value in enumerate(x):              yield (index, value)

現在，我們在原來的程式碼中調用這段個新的生成器函數：

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

其中，isinstance(value, (dict, list))，相當於isinstance(value, dict) or isinstance(value, list)。

我們來看一下運行效果：

掌握yield關鍵字，你的編程思路和想問題的方式會發生一個重大的轉變。