¿Cómo filtrar un diccionario según una función de condición arbitraria?
Tengo un diccionario de puntos, digamos:
>>> points={'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)}
Quiero crear un nuevo diccionario con todos los puntos cuyo valor xey es menor que 5, es decir, los puntos 'a', 'b' y 'd'.
Según el libro , cada diccionario tiene la items()función que devuelve una lista de (key, pair) tuplas:
>>> points.items()
[('a', (3, 4)), ('c', (5, 5)), ('b', (1, 2)), ('d', (3, 3))]
Entonces he escrito esto:
>>> for item in [i for i in points.items() if i[1][0]<5 and i[1][1]<5]:
... points_small[item[0]]=item[1]
...
>>> points_small
{'a': (3, 4), 'b': (1, 2), 'd': (3, 3)}
¿Existe una forma más elegante? Esperaba que Python tuviera alguna dictionary.filter(f)función súper impresionante...
Puedes usar un dict de comprensión:
{k: v for k, v in points.items() if v[0] < 5 and v[1] < 5}
Y en Python 2, a partir de 2.7:
{k: v for k, v in points.iteritems() if v[0] < 5 and v[1] < 5}
dict((k, v) for k, v in points.items() if all(x < 5 for x in v))
Puede optar por llamar .iteritems()en lugar de hacerlo .items()si está en Python 2 y pointspuede tener muchas entradas.
all(x < 5 for x in v)Puede ser excesivo si sabes con seguridad que cada punto siempre será solo 2D (en ese caso podrías expresar la misma restricción con un and) pero funcionará bien ;-).
points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items()))
