Comment filtrer un dictionnaire en fonction d'une fonction de condition arbitraire ?

Question

J'ai un dictionnaire de points, par exemple :

>>> points={'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)}

Je veux créer un nouveau dictionnaire avec tous les points dont la valeur x et y est inférieure à 5, c'est-à-dire les points 'a', 'b' et 'd'.

Selon le le livre chaque dictionnaire a le items() qui renvoie une liste de (key, pair) tuple :

>>> points.items()
[('a', (3, 4)), ('c', (5, 5)), ('b', (1, 2)), ('d', (3, 3))]

J'ai donc écrit ceci :

>>> 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-t-il un moyen plus élégant ? Je m'attendais à ce que Python dispose d'un systÃ?me de gestion de l'équilibre entre les deux. dictionary.filter(f) fonction...

Answer 1

Vous pouvez utiliser une dictée de compréhension :

{k: v for k, v in points.items() if v[0] < 5 and v[1] < 5}

Et dans Python 2, à partir de la version 2.7 :

{k: v for k, v in points.iteritems() if v[0] < 5 and v[1] < 5}

Answer 2

dict((k, v) for k, v in points.items() if all(x < 5 for x in v))

Vous pouvez choisir d'appeler .iteritems() au lieu de .items() si vous êtes en Python 2 et en points peut avoir un lot d'entrées.

all(x < 5 for x in v) peut s'avérer inutile si vous êtes certain que chaque point sera toujours uniquement en 2D (dans ce cas, vous pourriez exprimer la même contrainte avec une option and ) mais cela fonctionnera très bien;-).

Answer 3

points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items()))

Answer 4

dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)

Answer 5

dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)