Свести словарь словарей списков других словарей списков

Question

У меня есть это весьма вложенная словарь дерево:

sample = {"name": "one", 
      "id": "1", 
      "children": [{"name": "two", 
         "id": "2", 
         "children": [{"name": "six", 
             "id": "6", 
             "children": []}, 
            {"name": "seven", 
             "id": "7", 
             "children": []}]}, 
         {"name": "three", 
         "id": "3", 
         "children": []}, 
         {"name": "four", 
         "id": "4", 
         "children": []}, 
         {"name": "five", 
         "id": "5", 
         "children": []}]} 

Это просто пример, в действительности есть 7 или 8 уровней списков детей .. Также каждое имя и идентификатор уникальны.

Моя цель состоит в том, чтобы сгладить это дерево в словарь, который имеет значение всех ключей имен в качестве ключей и их идентификаторы в качестве второго ключа пары значений:

sample = {"one": {"id":"1"}, 
      "two": {"id":"2"}, 
      "three": {"id": "3"}, ...} 

На самом деле есть несколько пар ключ-значение, но Меня интересуют только имена и связанные с ними идентификаторы.

Я попытался обмотать голову, но мои навыки в рекурсиях не очень хорошие, было трудное время, поэтому я прошу вас о помощи. Также я искал аналогичные проблемы, но тот факт, что dicts инкапсулирован в списки, делает его не очень сопоставимым, для меня так или иначе ...

Я придумал решение для решения моей проблемы, но это был самый хакерский и уродливый код, который я каждый писал, и мне стыдно за это. В основном я преобразовал dict в его строковое представление и использовал regex, чтобы найти мои пары! И это плохо, но мне пришлось прототипировать что-то, чтобы успеть позаботиться о других проблемах ...

Итак, любые идеи, ребята?

Answer 1

Вы можете сделать рекурсивную функцию следующим образом (при условии, что каждый из dicts правильно структурировать):

def flatten(source, target): 
    target[source["name"]] = {"id": source["id"]} 
    for child in source["children"]: 
     flatten(child, target) 

Пример:

>>> d = {} 
>>> flatten(sample, d) 
>>> d 
{'seven': {'id': '7'}, 'six': {'id': '6'}, 'three': {'id': '3'}, 'two': {'id': '2'}, 'four': {'id': '4'}, 'five': {'id': '5'}, 'one': {'id': '1'}} 

Или, как это, если вам не нравится прохождение мишень ДИКТ в качестве параметра:

def flatten(source): 
    d = {source["name"]: {"id": source["id"]}} 
    for child in source["children"]: 
     d.update(flatten(child)) 
    return d 

Пример:

>>> flatten(sample) 
{'one': {'id': '1'}, 'four': {'id': '4'}, 'seven': {'id': '7'}, 'five': {'id': '5'}, 'six': {'id': '6'}, 'three': {'id': '3'}, 'two': {'id': '2'}} 

Можно также упростить вывод, чтобы быть простой, не вложенная словарь:

def flatten(source): 
    d = {source["name"]: source["id"]} 
    for child in source["children"]: 
     d.update(flatten(child)) 
    return d 

>>> flatten(sample) 
{'one': '1', 'four': '4', 'seven': '7', 'five': '5', 'six': '6', 'three': '3', 'two': '2'} 

Answer 2

Если ваши данные на самом деле более сложным, чем ваш пример:

def rec_get(d, k): 
    if isinstance(d, dict): 
     if k in d: 
      yield (d[k], {"id": d["id"]}) 
     for v in d.values(): 
      yield from rec_get(v, k) 
    elif isinstance(d, list): 
     for v in d: 
      yield from rec_get(v, k) 
print(dict(rec_get(sample ,"name"))) 

Выход:

{'five': {'id': '5'}, 'six': {'id': '6'}, 'four': {'id': '4'}, 'one': {'id': '1'}, 'three': {'id': '3'}, 'two': {'id': '2'}, 'seven': {'id': '7'}} 

Если вам нужна более общая функция, вы могли бы так повесят как:

from collections import OrderedDict 
from collections import Iterable 


def rec_get(d, **kwargs): 
    if isinstance(d, dict): 
     yield from ((d[k], k) for k in kwargs.keys() & d) 
     for v in d.values(): 
      yield from rec_get(v, **kwargs) 
    elif isinstance(d, Iterable) and not isinstance(d, str): 
     for v in d: 
      yield from rec_get(v, **kwargs) 

Который только в качестве примера, используя ввод и проходящие ключевые слова аргументы:

print(list(rec_get(sample, name="name", id="id"))) 

Выходы:

[('1', 'id'), ('one', 'name'), ('2', 'id'), ('two', 'name'), 
('6', 'id'), ('six', 'name'), ('7', 'id'), ('seven', 'name'), 
('3', 'id'), ('three', 'name'), ('4', 'id'), ('four', 'name'), ('5', 'id'), ('five', 'name')]