Как фильтровать словарь в соответствии с произвольной функцией условия?

У меня есть словарь точек, скажем:

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

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

Согласно книге, каждый словарь имеет функцию items(), которая возвращает список (key, pair) tuple:

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

Ответы

Ответ 1

В настоящее время в 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}

Ответ 2

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), но она будет работать нормально; -).

Ответ 3

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

Ответ 4

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

Ответ 5

Я думаю, что ответ Alex Martelli - это, безусловно, самый элегантный способ сделать это, но просто хотел добавить способ удовлетворить ваши потребности в супер-потрясающем методе dictionary.filter(f) в стиле 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)

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

Ответ 6

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

Ответ 7

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