Элегантный способ удаления полей из вложенных словарей

Question

Элегантный способ удаления полей из вложенных словарей

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

def delete_keys_from_dict(dict_del, lst_keys):
    """
    Delete the keys present in 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

Этот код работает, но он не очень элегантный, и я уверен, что есть лучшее решение.

20

python dictionary

задан fasouto 8 April 2018 в 20:55

2 ответа

Другие вопросы по тегам:

python dictionary

Похожие вопросы:

score 18 · 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

score 2 · Answer 2

Поскольку вам уже нужно перебрать каждый элемент в dict, я бы остановился на одном цикле и просто использовал набор для поиска ключей для удаления

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