Sprawdź, czy dany klucz już istnieje w słowniku i zwiększ go

Biorąc pod uwagę słownik, jak mogę się dowiedzieć, czy dany klucz w tym słowniku został już ustawiony na wartość inną niż Brak?

Tzn. Chcę to zrobić:

my_dict = {}

if (my_dict[key] != None):
  my_dict[key] = 1
else:
  my_dict[key] += 1

To znaczy, chcę zwiększyć wartość, jeśli już tam jest, lub ustawić ją na 1 w przeciwnym razie.

Szukasz collections.defaultdict(dostępne dla Python 2.5+). To

from collections import defaultdict

my_dict = defaultdict(int)
my_dict[key] += 1

zrobi co chcesz.

W przypadku zwykłych Pythonów dict, jeśli nie ma wartości dla danego klucza, nie dostaniesz się Nonepodczas uzyskiwania dostępu do słownika - KeyErrorzostanie podniesione. Więc jeśli chcesz używać zwykłego dict, zamiast kodu, którego byś użył

if key in my_dict:
    my_dict[key] += 1
else:
    my_dict[key] = 1

Wolę to zrobić w jednym wierszu kodu.

my_dict = {}

my_dict [some_key] = my_dict.get (some_key, 0) + 1

Słowniki mają funkcję get, która pobiera dwa parametry - żądany klucz i wartość domyślną, jeśli nie istnieje. Wolę tę metodę niż defaultdict, ponieważ chcesz obsłużyć przypadek, w którym klucz nie istnieje w tym jednym wierszu kodu, nie wszędzie.

Osobiście lubię używać setdefault()

my_dict = {}

my_dict.setdefault(some_key, 0)
my_dict[some_key] += 1

Potrzebujesz do tego key in dictidiomu.

if key in my_dict and not (my_dict[key] is None):
  # do something
else:
  # do something else

Jednak prawdopodobnie powinieneś rozważyć użycie defaultdict(jak sugeruje dF).

Aby odpowiedzieć na pytanie „ jak mogę się dowiedzieć, czy dany indeks w tym dykcie został już ustawiony na wartość inną niż Brak ”, wolałbym:

try:
  nonNone = my_dict[key] is not None
except KeyError:
  nonNone = False

Jest to zgodne z już przywołaną koncepcją EAFP (łatwiej prosić o wybaczenie niż o pozwolenie). Unika także podwójnego wyszukiwania klucza w słowniku, tak jak by to było w przypadkukey in my_dict and my_dict[key] is not None interesujące, gdyby wyszukiwanie było drogie.

W przypadku rzeczywistego problemu , który wystąpiłeś, tj. Zwiększenie wartości int, jeśli istnieje, lub ustawienie jej wartości domyślnej w innym przypadku, również polecam

my_dict[key] = my_dict.get(key, default) + 1

jak w odpowiedzi Andrew Wilkinsona.

Istnieje trzecie rozwiązanie, jeśli przechowujesz modyfikowalne obiekty w słowniku. Typowym tego przykładem jest multimapa , w której przechowujesz listę elementów dla swoich kluczy. W takim przypadku możesz użyć:

my_dict.setdefault(key, []).append(item)

Jeśli wartość klucza nie istnieje w słowniku, metoda setdefault ustawi drugi parametr setdefault. Zachowuje się jak standardowy my_dict [klucz], zwracając wartość klucza (która może być wartością nowo ustawioną).

Uzgodniony z Cgoldbergiem. Jak to robię to:

try:
    dict[key] += 1
except KeyError:
    dict[key] = 1

Więc albo zrób to jak wyżej lub użyj domyślnego dykta, jak sugerowali inni. Nie używaj instrukcji if. To nie jest Python.

Jak widać z wielu odpowiedzi, istnieje kilka rozwiązań. O jednej instancji LBYL (patrz przed skokiem) jeszcze nie wspomniano, metoda has_key ():

my_dict = {}

def add (key):
    if my_dict.has_key(key):
        my_dict[key] += 1
    else:
        my_dict[key] = 1

if __name__ == '__main__':
    add("foo")
    add("bar")
    add("foo")
    print my_dict

Sposób, w jaki próbujesz to zrobić, nazywa się LBYL (spójrz przed skokiem), ponieważ sprawdzasz warunki, zanim spróbujesz zwiększyć swoją wartość.

Inne podejście nazywa się EAFP (łatwiej prosić o wybaczenie niż o pozwolenie). W takim przypadku po prostu spróbuj wykonać operację (zwiększ wartość). Jeśli się nie powiedzie, wychwytujesz wyjątek i ustawiasz wartość na 1. Jest to nieco bardziej Pythoniczny sposób na zrobienie tego (IMO).

http://mail.python.org/pipermail/python-list/2003-May/205182.html

Trochę późno, ale to powinno zadziałać.

my_dict = {}
my_dict[key] = my_dict[key] + 1 if key in my_dict else 1

To nie jest bezpośrednia odpowiedź na pytanie, ale dla mnie wygląda na to, że możesz chcieć funkcjonalności kolekcji .

from collections import Counter

to_count = ["foo", "foo", "bar", "baz", "foo", "bar"]

count = Counter(to_count)

print(count)

print("acts just like the desired dictionary:")
print("bar occurs {} times".format(count["bar"]))

print("any item that does not occur in the list is set to 0:")
print("dog occurs {} times".format(count["dog"]))

print("can iterate over items from most frequent to least:")
for item, times in count.most_common():
    print("{} occurs {} times".format(item, times))

Daje to wynik

Counter({'foo': 3, 'bar': 2, 'baz': 1})
acts just like the desired dictionary:
bar occurs 2 times
any item that does not occur in the list is set to 0:
dog occurs 0 times
can iterate over items from most frequent to least:
foo occurs 3 times
bar occurs 2 times
baz occurs 1 times

Oto jeden linijka, którą ostatnio wymyśliłem, aby rozwiązać ten problem. Opiera się na metodzie słownika setdefault :

my_dict = {}
my_dict[key] = my_dict.setdefault(key, 0) + 1

Szukałem tego, nie znalazłem go w Internecie, a potem spróbowałem szczęścia z Try / Error i znalazłem

my_dict = {}

if my_dict.__contains__(some_key):
  my_dict[some_key] += 1
else:
  my_dict[some_key] = 1