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

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

>>> 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 будет иметь супер-awesome dictionary.filter(f) функцию …

7 Solutions collect form web for “Как фильтровать словарь в соответствии с произвольной функцией условия?”

В настоящее время в 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 (в этом случае вы можете выразить одно и то же ограничение с помощью a and ), но она будет работать нормально ;-).

 points_small = dict(filter(lambda (a,(b,c)): b<5 and c < 5, points.items())) 
 dict((k, v) for (k, v) in points.iteritems() if v[0] < 5 and v[1] < 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() то время как разрушающая итерация вызовет исключение.

 >>> 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) 
  • Есть ли быстрый способ генерировать букву алфавита в Python?
  • Содержание словаря для деструктурирования python-bind
  • Индекс по длине слова
  • Итерация через словарь Python с помощью клавиш в отсортированном порядке
  • В Python, как я могу получить следующий и предыдущий ключ: значение определенного ключа в словаре?
  • Python: использование vars () для назначения строки переменной
  • Альтернатива пониманию dict до Python 2.7
  • Как вы создаете вложенный dict в Python?
  • Сценарий Python для подсчета числа строк во всех файлах в каталоге
  • Как поднять ошибку, если дублирует ключи в словаре
  • наиболее эффективная структура данных для списка строк только для чтения (около 100 000) с быстрым префиксным поиском
  •  
    Interesting Posts for Van-Lav

    Разница между django-webtest и селеном

    pytest: как передать параметр класса в setup_class

    Найти новые сообщения, добавленные в почтовый ящик imap, так как я последний раз проверял с помощью python imaplib2?

    Django-Grappelli: Обратное для 'grp_related_lookup' с аргументами '()' и аргументами ключевого слова '{}' не найдено

    Почему отладка в eclipse / pydev настолько медленная для моей программы python?

    SQL-подобные функции окна в PANDAS: Нумерация строк в Python Pandas Dataframe

    Снять () нельзя интерпретировать?

    добавление двух последовательных номеров в список

    Flask Mega Tutorial – jinja2.exceptions.UndefinedError: «форма» не определена

    Чистый способ получить почти-LIFO поведение от многопроцессорности. Queue? (или даже просто * не * рядом с FIFO)

    удаление emojis из строки в Python

    Фильтровать список словарей

    Запуск m-файлов из Python

    Интерполяция в SciPy: поиск X, который производит Y

    jupyter работает под управлением ядра в разных env

    Python - лучший язык программирования в мире.