Vérifie si une clé donnée existe déjà dans un dictionnaire et l'incrémente

Question

Étant donné un dictionnaire, comment puis-je savoir si une clé donnée de ce dictionnaire a déjà été définie par une valeur autre que None ?

C'est-à-dire que je veux faire ça :

my_dict = {}

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

C'est-à-dire que je veux incrémenter la valeur s'il y en a déjà une, ou la mettre à 1 sinon.

Answer 1

Vous êtes à la recherche de collections.defaultdict (disponible pour Python 2.5+). Ce site

from collections import defaultdict

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

fera ce que vous voulez.

Pour le Python ordinaire dict s, s'il n'y a pas de valeur pour une clé donnée, vous allez no obtenir None lors de l'accès au dict -- un KeyError sera relevé. Donc, si vous voulez utiliser un dict Au lieu de votre code, vous utiliseriez

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

Answer 2

Je préfère faire cela en une seule ligne de code.

my\_dict = {}

my\_dict\[some\_key\] = my\_dict.get(some\_key, 0) + 1

Les dictionnaires ont une fonction, get, qui prend deux paramètres - la clé que vous voulez, et une valeur par défaut si elle n'existe pas. Je préfère cette méthode à defaultdict car vous ne voulez gérer que le cas où la clé n'existe pas dans cette ligne de code, pas partout.

Answer 3

Vous avez besoin de la key in dict idiome pour cela.

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

Cependant, vous devriez probablement envisager d'utiliser defaultdict (comme dF l'a suggéré).

Answer 4

D'accord avec cgoldberg. Voici comment je procède :

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

Donc, soit vous le faites comme ci-dessus, soit vous utilisez un dict par défaut comme d'autres l'ont suggéré. N'utilisez pas d'instructions if. Ce n'est pas Python.

Answer 5

Comme vous pouvez le voir dans les nombreuses réponses, il existe plusieurs solutions. Un exemple de LBYL (look before you leap) n'a pas encore été mentionné, la méthode 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