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

在上一篇文章裏面，咱們講到了如何使用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關鍵字，你的編程思路和想問題的方式會發生一個重大的轉變。