¿Cómo agregar o incrementar una entrada de diccionario?

Question

Actualmente estoy volviendo a comprometerme con Python después de una larga ausencia y me encanta. Sin embargo, me encuentro encontrando un patrón una y otra vez. Sigo pensando que debe haber una mejor manera de expresar lo que quiero y que probablemente lo estoy haciendo mal.

El código que estoy escribiendo es de la siguiente forma:

# foo is a dictionary 
if foo.has_key(bar): 
    foo[bar] += 1 
else: 
    foo[bar] = 1 

estoy escribiendo esto mucho en mis programas. Mi primera reacción es llevarlo a una función auxiliar, pero a menudo las bibliotecas de Python ya ofrecen cosas como esta.

¿Hay algún pequeño truco de sintaxis que me falta? ¿O es así como debería hacerse?

Answer 1

Utilice un defaultdict:

from collections import defaultdict 

foo = defaultdict(int) 
foo[bar] += 1 

En Python> = 2.7, también tiene una clase separada Counter para estos fines. Para Python 2.5 y 2.6, puede usar su backported version.

Answer 2

Para Python> = 2.5 puede hacer lo siguiente:

foo[bar] = 1 if bar not in foo else foo[bar]+1 

Answer 3

También puede tomar ventaja de la estructura de control en el manejo de excepciones. Un KeyError excepción es lanzada por un diccionario cuando intenta asignar un valor a una clave inexistente:

my_dict = {} 
try: 
    my_dict['a'] += 1 
except KeyError, err: # in 2.6: `except KeyError as err:` 
    my_dict['a'] = 1 

Answer 4

get() El método dict 's toma un segundo parámetro opcional que se puede utilizar para proporcionar un valor predeterminado si la clave solicitada no se encuentra:

foo[bar] = foo.get(bar, 0) + 1 

Answer 5

Hice algunas comparaciones de tiempo. Más o menos igual. Sin embargo, el comando one-lined .get() es el más rápido.

Salida:

get 0.543551800627 
exception 0.587318710994 
haskey 0.598421703081 

Código:

import timeit 
import random 

RANDLIST = [random.randint(0, 1000) for i in range(10000)] 

def get(): 
    foo = {} 
    for bar in RANDLIST: 
     foo[bar] = foo.get(bar, 0) + 1 


def exception(): 
    foo = {} 
    for bar in RANDLIST: 
     try: 
      foo[bar] += 1 
     except KeyError: 
      foo[bar] = 1 


def haskey(): 
    foo = {} 
    for bar in RANDLIST: 
     if foo.has_key(bar): 
      foo[bar] += 1 
     else: 
      foo[bar] = 1 


def main(): 
    print 'get', timeit.timeit('get()', 'from __main__ import get', number=100) 
    print 'exception', timeit.timeit('exception()', 'from __main__ import exception', number=100) 
    print 'haskey', timeit.timeit('haskey()', 'from __main__ import haskey', number=100) 


if __name__ == '__main__': 
    main()