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

Я имею словарь точек, говорю:

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

Я хочу создать новый словарь со всеми точками, значение X и Y которых меньше, чем 5, т.е. указывает, '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) функция...

190
задан Ciro Santilli 新疆改造中心法轮功六四事件 20 November 2015 в 12:23
поделиться

4 ответа

dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)
6
ответ дан 23 November 2019 в 05:35
поделиться
dict((k, v) for k, v in points.items() if all(x < 5 for x in v))

Вы можете позвонить по адресу .iteritems () вместо .items () , если вы в Python 2 и точки могут иметь много записей.

all (x <5 для x в v) может оказаться излишним, если вы точно знаете, что каждая точка всегда будет только 2D (в этом случае вы можете выразить то же ограничение с помощью и ) но работать будет нормально ;-).

108
ответ дан 23 November 2019 в 05:35
поделиться
points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items()))
20
ответ дан 23 November 2019 в 05:35
поделиться
dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 5)
9
ответ дан 23 November 2019 в 05:35
поделиться
Другие вопросы по тегам:

Похожие вопросы: