Elegancki sposób usuwania pól ze słowników zagnieżdżonych

Question

Musiałem usunąć niektóre pola ze słownika, klucze tych pól znajdują się na liście. Więc piszę tę funkcję:

def delete_keys_from_dict(dict_del, lst_keys): 
    """ 
    Delete the keys present in the lst_keys from the dictionary. 
    Loops recursively over nested dictionaries. 
    """ 
    dict_foo = dict_del.copy()#Used as iterator to avoid the 'DictionaryHasChanged' error 
    for field in dict_foo.keys(): 
     if field in lst_keys: 
      del dict_del[field] 
     if type(dict_foo[field]) == dict: 
      delete_keys_from_dict(dict_del[field], lst_keys) 
    return dict_del 

Ten kod działa, ale nie jest zbyt elegancki i jestem pewien, że możesz napisać lepsze rozwiązanie.

Answer 1

def delete_keys_from_dict(dict_del, lst_keys): 
    for k in lst_keys: 
     try: 
      del dict_del[k] 
     except KeyError: 
      pass 
    for v in dict_del.values(): 
     if isinstance(v, dict): 
      delete_keys_from_dict(v, lst_keys) 

    return dict_del 

Answer 2

Skoro już trzeba pętli każdy element w dict, będę trzymać się z jednej pętli i po prostu upewnij się, aby użyć zestawu dla patrząc klucze do usunięcia

def delete_keys_from_dict(dict_del, the_keys): 
    """ 
    Delete the keys present in the lst_keys from the dictionary. 
    Loops recursively over nested dictionaries. 
    """ 
    # make sure the_keys is a set to get O(1) lookups 
    if type(the_keys) is not set: 
     the_keys = set(the_keys) 
    for k,v in dict_del.items(): 
     if k in the_keys: 
      del dict_del[k] 
     if isinstance(v, dict): 
      delete_keys_from_dict(v, the_keys) 
    return dict_del 

Answer 3

Ponieważ pytanie wymagało eleganckiego sposobu, przekażę moje rozwiązanie ogólnego przeznaczenia do splatania struktur zagnieżdżonych. Po pierwsze, należy zainstalować boltons utility package z pip install boltons, a następnie:

from boltons.iterutils import remap 

data = {'one': 'remains', 'this': 'goes', 'of': 'course'} 
bad_keys = set(['this', 'is', 'a', 'list', 'of', 'keys']) 

drop_keys = lambda path, key, value: key not in bad_keys 
clean = remap(data, visit=drop_keys) 
print(clean) 

# Output: 
{'one': 'remains'} 

W skrócie, the remap utility jest w pełni funkcjonalny, ale zwięzły podejście do obsługi struktur danych rzeczywistych, które często są zagnieżdżone, a nawet może zawierać cykle i specjalne pojemniki.

This page ma wiele innych przykładów, w tym te, które pracują z większymi obiektami z API Githuba.

Jest to czysty Python, więc działa wszędzie i jest w pełni przetestowany w Pythonie 2.7 i 3.3+. Najlepszy ze wszystkich, napisałem to dla dokładnie takich przypadków, więc jeśli znajdziesz skrzynkę, której nie obsługuje, możesz mnie zaalarmować, aby to naprawić right here.

Answer 4

Korzystanie niesamowite kod z this posta i dodać niewielką stwierdzenie:

def remove_fields(self, d, list_of_keys_to_remove): 
     if not isinstance(d, (dict, list)): 
      return d 
     if isinstance(d, list): 
      return [v for v in (self.remove_fields(v, list_of_keys_to_remove) for v in d) if v] 
     return {k: v for k, v in ((k, self.remove_fields(v, list_of_keys_to_remove)) for k, v in d.items()) if k not in list_of_keys_to_remove} 

Answer 5

myślę, że jest bardziej elegancki następujący:

def delete_keys_from_dict(dict_del, lst_keys): 
    if not isinstance(dict_del, dict): 
     return dict_del 
    return {key:value for key,value in ((key, delete_keys_from_dict(value)) for key,value in dict_del.items()) if key not in lst_keys}