Comment filtrer un dictionnaire selon une fonction de condition arbitraire?

J'ai un dictionnaire de points, disons:

>>> 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 livre , chaque dictionnaire a la items()fonction, qui retourne 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 une manière plus élégante? Je m'attendais à ce que Python ait une fonction super géniale dictionary.filter(f)...

De nos jours, en Python 2.7 et supérieur, vous pouvez utiliser une compréhension de dict:

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

Et en Python 3:

{k: v for k, v in points.items() 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))

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

all(x < 5 for x in v)peut être exagéré si vous savez avec certitude que chaque point sera toujours 2D uniquement (dans ce cas, vous pourriez exprimer la même contrainte avec un and) mais cela fonctionnera bien ;-).

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

>>> points = {'a': (3, 4), 'c': (5, 5), 'b': (1, 2), 'd': (3, 3)}
>>> dict(filter(lambda x: (x[1][0], x[1][1]) < (5, 5), points.items()))

{'a': (3, 4), 'b': (1, 2), 'd': (3, 3)}

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

Je pense que la réponse d'Alex Martelli est certainement la façon la plus élégante de le faire, mais je voulais juste ajouter un moyen de satisfaire votre besoin d'une dictionary.filter(f)méthode super géniale d'une manière Pythonique:

class FilterDict(dict):
    def __init__(self, input_dict):
        for key, value in input_dict.iteritems():
            self[key] = value
    def filter(self, criteria):
        for key, value in self.items():
            if (criteria(value)):
                self.pop(key)

my_dict = FilterDict( {'a':(3,4), 'b':(1,2), 'c':(5,5), 'd':(3,3)} )
my_dict.filter(lambda x: x[0] < 5 and x[1] < 5)

Fondamentalement, nous créons une classe qui hérite de dict, mais ajoute la méthode de filtrage. Nous devons utiliser .items()pour le filtrage, car l'utilisation .iteritems()lors d'une itération destructrice soulèvera une exception.

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