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

Étant donné un dictionnaire, comment puis-je savoir si une clé donnée dans ce dictionnaire a déjà été définie sur une valeur non nulle?

C'est-à-dire que je veux faire ceci:

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.

Vous recherchez collections.defaultdict(disponible pour Python 2.5+). Ce

from collections import defaultdict

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

fera ce que vous voulez.

Pour Python régulière dicts, s'il n'y a pas de valeur pour une clé donnée, vous n'obtenez lorsque vous accédez au dict - un ressusciteront. Donc, si vous souhaitez utiliser un régulier , au lieu de votre code, vous utiliseriezNoneKeyErrordict

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

Je préfère le faire dans 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 seule ligne de code, pas partout.

J'aime personnellement utiliser setdefault()

my_dict = {}

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

Vous avez besoin de l' key in dictidiome 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 suggéré par dF).

Pour répondre à la question " comment puis-je savoir si un index donné dans ce dict a déjà été défini sur une valeur non nulle ", je préférerais ceci:

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

Cela est conforme au concept déjà invoqué d'EAFP (plus facile de demander pardon puis permission). Cela évite également la recherche de clé en double dans le dictionnaire comme ce serait le cas dans key in my_dict and my_dict[key] is not Nonece qui est intéressant si la recherche est coûteuse.

Pour le problème réel que vous avez posé, c'est-à-dire incrémenter un int s'il existe, ou le définir sur une valeur par défaut sinon, je recommande également le

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

comme dans la réponse d'Andrew Wilkinson.

Il existe une troisième solution si vous stockez des objets modifiables dans votre dictionnaire. Un exemple courant pour cela est un multimap , où vous stockez une liste d'éléments pour vos clés. Dans ce cas, vous pouvez utiliser:

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

Si une valeur pour key n'existe pas dans le dictionnaire, la méthode setdefault la définira sur le deuxième paramètre de setdefault. Il se comporte exactement comme un my_dict [clé] standard, renvoyant la valeur de la clé (qui peut être la nouvelle valeur définie).

D'accord avec cgoldberg. Comment je fais:

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

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

Comme vous pouvez le voir dans les nombreuses réponses, il existe plusieurs solutions. Une instance de LBYL (regardez avant de sauter) n'a pas encore été mentionnée, 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

La façon dont vous essayez de le faire s'appelle LBYL (regardez avant de sauter), car vous vérifiez les conditions avant d'essayer d'augmenter votre valeur.

L'autre approche est appelée EAFP (plus facile de demander pardon puis permission). Dans ce cas, vous essayez simplement l'opération (incrémentez la valeur). S'il échoue, vous interceptez l'exception et définissez la valeur sur 1. Il s'agit d'une façon légèrement plus Pythonique de le faire (IMO).

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

Un peu tard mais ça devrait marcher.

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

Cela ne répond pas directement à la question, mais pour moi, il semble que vous souhaitiez peut-être la fonctionnalité des collections .

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))

Cela se traduit par la sortie

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

Voici une ligne que j'ai récemment trouvée pour résoudre ce problème. Il est basé sur la méthode du dictionnaire setdefault :

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

Je le cherchais, je ne l'ai pas trouvé sur le web puis j'ai tenté ma chance avec Try / Error et je l'ai trouvé

my_dict = {}

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