Як відфільтрувати словник за функцією довільної умови?

У мене є словник балів, скажімо:

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

Я хочу створити новий словник з усіма точками, значення x і y менше 5, тобто точки 'a', 'b' і 'd'.

Відповідно до книги , кожен словник має items()функцію, яка повертає список (key, pair) кортежу:

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

Тому я написав це:

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

Чи є більш елегантний спосіб? Я очікував, що Python матиме якусь надзвичайну dictionary.filter(f)функцію ...

Сьогодні в Python 2.7 і новіших версіях ви можете використовувати розуміння dict:

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

А в 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))

Ви можете вибрати дзвінок .iteritems()замість того, .items()якщо ви перебуваєте на Python 2 і, pointsможливо, є багато записів.

all(x < 5 for x in v)може бути зайвим, якщо ви точно знаєте, кожна точка завжди буде лише 2D (у такому випадку ви можете висловити одне і те ж обмеження and), але це буде добре ;-).

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)

Я думаю, що відповідь Алекса Мартеллі - це, безумовно, найелегантніший спосіб зробити це, але просто хотілося додати спосіб задоволення вашого бажання надзвичайно дивовижним dictionary.filter(f)методом в пітонічному вигляді:

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)

В основному ми створюємо клас, який успадковує dict, але додає метод фільтра. Нам потрібно використовувати .items()для фільтрації, оскільки використання .iteritems()під час деструктивної ітерації призведе до виключення.

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