Extracción de un subconjunto de un diccionario de dentro de una lista

Question

Esto es realmente sólo es fácil de explicar con un ejemplo, para quitar la intersección de una lista desde el interior de un diccionario que suelo hacer algo como esto:

a = {1:'', 2:'', 3:'', 4:''} 
exclusion = [3, 4, 5] 

# have to build up a new list or the iteration breaks 
toRemove = [] 
for var in a.iterkeys(): 
    if var in exclusion: 
     toRemove.append(var) 

for var in toRemove: 
    del a[var] 

Esto podría parecer un ejemplo inusual, pero es sorprendente la cantidad de veces que he tenido que hacer algo como esto. Hacer esto con conjuntos sería mucho mejor, pero claramente quiero retener los 'valores' para el dict.

Este método es molesto porque requiere dos bucles y una matriz adicional. ¿Hay una manera más limpia y más eficiente de hacer esto?

Answer 1

Considere dict.pop:

for key in exclusion: 
    a.pop(key, None) 

El None mantiene pop de lanzar una excepción cuando key no es una clave.

Answer 2

¿Por qué no usar el método keys, en lugar de iterkeys? De esta forma, puede hacerlo en un bucle porque devuelve una lista, no un iterador.

Answer 3

a = dict((key,value) for (key,value) in a.iteritems() if key not in exclusion) 

Answer 4

Puede cambiar su lista de exclusión a un conjunto, luego simplemente use intersección para obtener la superposición.

exclusion = set([3, 4, 5]) 

for key in exclusion.intersection(a): 
    del a[key]