¿Cómo se compara el rendimiento de las búsquedas de claves del diccionario en Python?

Question

¿Cómo:

dict = {} 
if key not in dict: 
dict[key] = foo 

Compare con:

try: 
dict[key] 
except KeyError: 
dict[key] = foo 

es decir, es el aspecto de una llave de ninguna manera más rápida que la búsqueda lineal a través dict.keys(), que supongo que la primera forma va a hacer?

Answer 1

La respuesta depende de la frecuencia con la tecla ya está en el dict (Por cierto, ¿alguien ha mencionado a lo mal que una idea es ocultar una orden interna como dict detrás de una variable?)

if key not in dct: 
dct[key] = foo 

Si la clave está en el diccionario, esta hace una búsqueda de diccionario. Si la clave está en el diccionario, busca el diccionario dos veces.

try: 
dct[key] 
except KeyError: 
dct[key] = foo 

Esto puede ser un poco más rápido para el caso en que la clave está en el diccionario, pero lanzar una excepción tiene todo un gran sobrecarga, por lo que no es casi siempre la mejor opción.

dct.setdefault(key, foo) 

Ésta es un poco difícil: siempre implica dos búsquedas de diccionario: el primero es encontrar el método setdefault en la clase dict, la segunda es la búsqueda de key en el objeto dct. Además, si foo es una expresión, se evaluará siempre, mientras que las opciones anteriores solo la evaluarán cuando sea necesario.

Consulte también collections.defaultdict. Esa es la solución más adecuada para una gran clase de situaciones como esta.

Answer 2

my_dict.get(key, foo) devuelve foo si la clave no está en my_dict. El valor predeterminado es Ninguno, por lo que my_dict.get(key) devolverá None si la clave no está en my_dict. La primera de tus opciones es mejor si solo quieres agregar la clave a tu diccionario. No te preocupes por la velocidad aquí. Si encuentra que llenar su diccionario es un punto caliente en su programa, entonces piénselo. Pero no lo es. Entonces no.

Answer 3

Probar: my_dict.setdefault(key, default). Sin embargo, es un poco más lento que las otras opciones.

Si key es en el diccionario, devuelva su valor. Si no, inserte key con un valor de default y devuelva default. default está predeterminado a Ninguno.

#!/usr/bin/env python 

example_dict = dict(zip(range(10), range(10))) 

def kn(key, d): 
    if key not in d: 
     d[key] = 'foo' 

def te(key, d): 
    try: 
     d[key] 
    except KeyError: 
     d[key] = 'foo' 

def sd(key, d): 
    d.setdefault(key, 'foo') 

if __name__ == '__main__': 
    from timeit import Timer 

    t = Timer("kn(2, example_dict)", "from __main__ import kn, example_dict") 
    print t.timeit() 
    t = Timer("te(2, example_dict)", "from __main__ import te, example_dict") 
    print t.timeit() 
    t = Timer("sd(2, example_dict)", "from __main__ import sd, example_dict") 
    print t.timeit() 

    # kn: 0.249855041504 
    # te: 0.244259119034 
    # sd: 0.375113964081 

Answer 4

Usted está buscando el método setdefault:

>>> r = {} 
>>> r.setdefault('a', 'b') 
'b' 
>>> r 
{'a': 'b'} 
>>> r.setdefault('a', 'e') 
'b' 
>>> r 
{'a': 'b'} 

Answer 5

sólo para aclarar un punto: if key not in d no hacer una búsqueda lineal a través de teclas d's. Utiliza la tabla hash de dict para encontrar rápidamente la clave.