Come filtrare un dizionario in base a una funzione di condizione arbitraria?

Ho un dizionario di punti, ad esempio:

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

Voglio creare un nuovo dizionario con tutti i punti il ​​cui valore xey è inferiore a 5, ovvero i punti 'a', 'b' e 'd'.

Secondo il libro , ogni dizionario ha la items()funzione, che restituisce un elenco di (key, pair) tuple:

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

Quindi ho scritto questo:

>>> 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)}

C'è un modo più elegante? Mi aspettavo che Python avesse una funzione super fantastica dictionary.filter(f)...

Al giorno d'oggi, in Python 2.7 e versioni successive, è possibile utilizzare una comprensione dict:

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

E in 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))

Puoi scegliere di chiamare .iteritems()invece .items()che in Python 2 e pointspotresti avere molte voci.

all(x < 5 for x in v)può essere eccessivo se sai per certo che ogni punto sarà sempre solo 2D (in tal caso potresti esprimere lo stesso vincolo con un and) ma funzionerà bene ;-).

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)

Penso che la risposta di Alex Martelli sia sicuramente il modo più elegante per farlo, ma volevo solo aggiungere un modo per soddisfare il tuo desiderio di un dictionary.filter(f)metodo super fantastico in un modo Pythonic:

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)

Fondamentalmente creiamo una classe che eredita da dict, ma aggiunge il metodo di filtro. Dobbiamo utilizzare .items()per il filtro, poiché l'utilizzo .iteritems()durante l'iterazione distruttiva genererà un'eccezione.

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