Как сгладить некоторые уровни вложенного словаря? - PullRequest
Новая
       6

Как сгладить некоторые уровни вложенного словаря?

0 голосов
/ 04 марта 2020

У меня есть вложенный словарь d, как показано ниже 

d = {'id': {"a,b": {'id': {"x": {'id': None},
                           "y": {'id': {"a": {'id': None},
                                        "b": {'id': None}}}}},
            "c,d": {'id': {"c": {'id': None},
                            "d": {'id': {"x": {'id': None},
                                         "y": {'id': None}}}}}}}

, и я хотел бы удалить некоторые уровни и сжать его до следующего результата: 

{"a,b": {"x": None,
         "y": {"a": None,
               "b": None}},
 "c,d": {"c": None,
         "d": {"x": None,
               "y": None}}}

Хотите удалить вложенные слова словарь с ключом id и замените его внутренним словарем

Моя отправная точка: 

def unnest_dictionary(d):
    for k,v in d.items():
        if isinstance(v, dict):
            unnest_dictionary(v)
        if k=='id':
            ......

Не уверен, как от него откатить

1 Ответ

1 голос
/ 05 марта 2020

Вот как я решил это за

Я сгладил словарь, удалил уровни с помощью id, а затем снова вложил его 

import re

d = {'id': {"a,b": {'id': {"x": {'id': None},
                           "y": {'id': {"a": {'id': None},
                                        "b": {'id': None}}}}},
            "c,d": {'id': {"c": {'id': None},
                            "d": {'id': {"x": {'id': None},
                                         "y": {'id': None}}}}}}}


def flatten_dict(dd, separator ='_', prefix =''): 
    return { prefix + separator + k if prefix else k : v 
             for kk, vv in dd.items() 
             for k, v in flatten_dict(vv, separator, kk).items() 
             } if isinstance(dd, dict) else { prefix : dd }

def nest_dict(dict1): 
    result = {} 
    for k, v in dict1.items(): 

        split_rec(k, v, result) 
    return result 

def split_rec(k, v, out): 

    k, *rest = k.split('_', 1) 
    if rest: 
        split_rec(rest[0], v, out.setdefault(k, {})) 
    else: 
        out[k] = v  

flat_d = flatten_dict(d)
for k in list(flat_d.keys()):
    new_key = re.sub(r'_id|id_','',k)
    flat_d[new_key] = flat_d.pop(k)
nested_d = nest_dict(flat_d)

print(nested_d)
# {'a,b': {'x': None, 'y': {'a': None, 'b': None}}, 'c,d': {'c': None, 'd': {'x': None, 'y': None}}}
...