Насколько легко свести вложенные словари с yield ?

Python

Мы часто сталкиваемся с данными различных словарных наборов словарей, таких как:

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ключевое слово для решения смешанной вложенности словарей и списков.

Категории