Блог

У меня есть список объектов, и я хочу отфильтровать список таким образом, чтобы в результате было только одно вхождение каждого значения атрибута.

Допустим, у меня есть три объекта

 obj1.my_attr = 'a'
obj2.my_attr = 'b'
obj3.my_attr = 'b'

obj_list = [obj1, obj2, obj3]
  

И, и в конце, я хочу получить [obj1, obj2] . На самом деле порядок не имеет значения, поэтому [obj1, obj3] он точно такой же хороший.

Сначала я подумал о типичных императивных неуклюжих способах, таких как следующие:

 record = set()
result = []

for obj in obj_list:
    if obj.my_attr not in record:
        record.add(obj.my_attr)
        result.append(obj)
  

Затем я подумал о том, чтобы сопоставить его со словарем, использовать ключ для переопределения любой предыдущей записи и, наконец, извлечь значения:

 result = {obj.my_attr: obj for obj in obj_list}.values() 
  

Этот вариант выглядит неплохо, но я хотел бы знать, существует ли какой-либо более элегантный, эффективный или функциональный способ достижения этой цели. Может быть, какая-нибудь приятная штука, скрытая в стандартной библиотеке… Заранее спасибо.

Если вы хотите использовать функциональный стиль программирования на Python, вы можете ознакомиться с пакетом toolz. С toolz вы могли бы просто сделать:

 toolz.unique(obj_list, key=lambda x: x.my_attr)
  

Для повышения производительности вы могли бы использовать operator.attrgetter('my_attr') вместо лямбда-функции для ключа. Вы также могли бы использовать cytoolz, который представляет собой быструю реализацию toolz , написанную на Cython.

Вы могли бы использовать объект, который определял бы пользовательскую __hash__ функцию:

 class HashMyAttr:
    def __init__(self, obj):
        self.obj = obj
    def __hash__(self):
        return self.obj.my_attr.__hash__()
    def __eq__(self, other):
         return self.obj.my_attr == other.obj.my_attr
  

И используйте его как:

 obj_list = [x.obj for x in set(HashMyAttr(obj) for obj in obj_list)]