Bezpieczne usuwanie wielu kluczy ze słownika

Wiem, aby bezpiecznie usunąć wpis, „klucz” z mojego słownika d, robisz to:

if d.has_key('key'):
    del d['key']

Jednak muszę bezpiecznie usunąć wiele wpisów ze słownika. Myślałem o zdefiniowaniu wpisów w krotce, ponieważ będę musiał to zrobić więcej niż raz.

entitiesToREmove = ('a', 'b', 'c')
for x in entitiesToRemove:
    if d.has_key(x):
        del d[x]

Zastanawiałem się jednak, czy istnieje mądrzejszy sposób, aby to zrobić?

Dlaczego nie tak:

entries = ('a', 'b', 'c')
the_dict = {'b': 'foo'}

def entries_to_remove(entries, the_dict):
    for key in entries:
        if key in the_dict:
            del the_dict[key]

Bardziej zwarta wersja została dostarczona przez mattbornskiego za pomocą dict.pop ()

d = {'some':'data'}
entriesToRemove = ('any', 'iterable')
for k in entriesToRemove:
    d.pop(k, None)

Korzystanie ze zrozumiałych słów

final_dict = {key: t[key] for key in t if key not in [key1, key2]}

gdzie klucz1 i klucz2 mają zostać usunięte.

W poniższym przykładzie klucze „b” i „c” mają zostać usunięte i są przechowywane na liście kluczy.

>>> a
{'a': 1, 'c': 3, 'b': 2, 'd': 4}
>>> keys = ["b", "c"]
>>> print {key: a[key] for key in a if key not in keys}
{'a': 1, 'd': 4}
>>> 

rozwiązaniem jest użycie mapifilter funkcjonuje

python 2

d={"a":1,"b":2,"c":3}
l=("a","b","d")
map(d.__delitem__, filter(d.__contains__,l))
print(d)

python 3

d={"a":1,"b":2,"c":3}
l=("a","b","d")
list(map(d.__delitem__, filter(d.__contains__,l)))
print(d)

dostajesz:

{'c': 3}

Jeśli potrzebujesz również pobrać wartości usuwanych kluczy, byłby to całkiem dobry sposób:

valuesRemoved = [d.pop(k, None) for k in entitiesToRemove]

Oczywiście możesz to zrobić tylko po to, aby usunąć klucze z d, ale niepotrzebnie utworzyłbyś listę wartości za pomocą funkcji list. Trochę niejasne jest również użycie funkcji rozumienia listy tylko dla efektu ubocznego funkcji.

Znalazłem rozwiązanie za pomocą popimap

d = {'a': 'valueA', 'b': 'valueB', 'c': 'valueC', 'd': 'valueD'}
keys = ['a', 'b', 'c']
list(map(d.pop, keys))
print(d)

Wynik tego:

{'d': 'valueD'}

Odpowiedziałem na to pytanie tak późno, ponieważ myślę, że w przyszłości pomoże to, jeśli ktoś będzie szukał tego samego. I to może pomóc.

Aktualizacja

Powyższy kod spowoduje wyświetlenie błędu, jeśli klucz nie istnieje w dict.

DICTIONARY = {'a': 'valueA', 'b': 'valueB', 'c': 'valueC', 'd': 'valueD'}
keys = ['a', 'l', 'c']

def remove_keys(key):
    try:
        DICTIONARY.pop(key, None)
    except:
        pass  # or do any action

list(map(remove_key, keys))
print(DICTIONARY)

wynik:

DICTIONARY = {'b': 'valueB', 'd': 'valueD'}

Nie mam problemu z żadną z istniejących odpowiedzi, ale zdziwiłem się, że nie znalazłem tego rozwiązania:

keys_to_remove = ['a', 'b', 'c']
my_dict = {k: v for k, v in zip("a b c d e f g".split(' '), [0, 1, 2, 3, 4, 5, 6])}

for k in keys_to_remove:
    try:
        del my_dict[k]
    except KeyError:
        pass

assert my_dict == {'d': 3, 'e': 4, 'f': 5, 'g': 6}

Uwaga: natknąłem się na to pytanie, przychodząc stąd . A moja odpowiedź jest związana z tą odpowiedzią .

Dlaczego nie:

entriestoremove = (2,5,1)
for e in entriestoremove:
    if d.has_key(e):
        del d[e]

Nie wiem, co masz na myśli mówiąc „mądrzejszy sposób”. Z pewnością istnieją inne sposoby, być może ze słownikiem:

entriestoremove = (2,5,1)
newdict = {x for x in d if x not in entriestoremove}

inline

import functools

#: not key(c) in d
d = {"a": "avalue", "b": "bvalue", "d": "dvalue"}

entitiesToREmove = ('a', 'b', 'c')

#: python2
map(lambda x: functools.partial(d.pop, x, None)(), entitiesToREmove)

#: python3

list(map(lambda x: functools.partial(d.pop, x, None)(), entitiesToREmove))

print(d)
# output: {'d': 'dvalue'}

Niektóre testy czasowe dla cpythona 3 pokazują, że prosta pętla for jest najszybszym sposobem i jest całkiem czytelna. Dodanie funkcji również nie powoduje dużego obciążenia:

wyniki timeit (10 tys. iteracji):

• all(x.pop(v) for v in r) # 0.85
• all(map(x.pop, r)) # 0.60
• list(map(x.pop, r)) # 0.70
• all(map(x.__delitem__, r)) # 0.44
• del_all(x, r) # 0.40
• <inline for loop>(x, r) # 0.35

def del_all(mapping, to_remove):
      """Remove list of elements from mapping."""
      for key in to_remove:
          del mapping[key]

W przypadku małych iteracji robienie tego „inline” było nieco szybsze z powodu narzutu wywołania funkcji. Ale del_alljest bezpieczny, wielokrotnego użytku i szybszy niż wszystkie konstrukcje rozumienia i mapowania Pythona.

Myślę, że użycie faktu, że klucze można traktować jako zestaw, jest najmilszym sposobem, jeśli korzystasz z Pythona 3:

def remove_keys(d, keys):
    to_remove = set(keys)
    filtered_keys = d.keys() - to_remove
    filtered_values = map(d.get, filtered_keys)
    return dict(zip(filtered_keys, filtered_values))

Przykład:

>>> remove_keys({'k1': 1, 'k3': 3}, ['k1', 'k2'])
{'k3': 3}

Byłoby miło mieć pełne wsparcie dla metod ustawiania dla słowników (a nie przeklętego bałaganu, który dostajemy w Pythonie 3.9), abyś mógł po prostu „usunąć” zestaw kluczy. Jeśli jednak tak nie jest i masz duży słownik z potencjalnie dużą liczbą kluczy do usunięcia, możesz chcieć wiedzieć o wydajności. Stworzyłem więc kod, który tworzy coś wystarczająco dużego do znaczących porównań: macierz 100 000 x 1000, czyli łącznie 10 000,00 elementów.

from itertools import product
from time import perf_counter

# make a complete worksheet 100000 * 1000
start = perf_counter()
prod = product(range(1, 100000), range(1, 1000))
cells = {(x,y):x for x,y in prod}
print(len(cells))

print(f"Create time {perf_counter()-start:.2f}s")
clock = perf_counter()
# remove everything above row 50,000

keys = product(range(50000, 100000), range(1, 100))

# for x,y in keys:
#     del cells[x, y]

for n in map(cells.pop, keys):
    pass

print(len(cells))
stop = perf_counter()
print(f"Removal time {stop-clock:.2f}s")

10 milionów lub więcej elementów nie jest niczym niezwykłym w niektórych sytuacjach. Porównując dwie metody na moim komputerze lokalnym, widzę niewielką poprawę podczas używania mapi popprawdopodobnie z powodu mniejszej liczby wywołań funkcji, ale obie zajmują około 2,5 sekundy na moim komputerze. Ale to blednie w porównaniu z czasem potrzebnym do utworzenia słownika w pierwszej kolejności (55 s) lub z uwzględnieniem sprawdzeń w pętli. Jeśli jest to prawdopodobne, najlepiej jest utworzyć zestaw będący przecięciem kluczy słownika i filtru:

keys = cells.keys() & keys

Podsumowując: deljest już mocno zoptymalizowany, więc nie martw się o jego używanie.

Spóźniłem się na tę dyskusję, ale dla kogokolwiek innego. Rozwiązaniem może być utworzenie listy kluczy jako takiej.

k = ['a','b','c','d']

Następnie użyj pop () w składaniu listowym lub pętli for, aby iterować po klawiszach i wyskakiwać pojedynczo.

new_dictionary = [dictionary.pop(x, 'n/a') for x in k]

Wartość „n / a” oznacza, że ​​klucz nie istnieje, należy zwrócić wartość domyślną.