Сопоставимые классы в Python 3

Чтобы сделать классы сопоставимыми, вам нужно только реализовать __lt__ и украсить класс functools.total_ordering. Вы также должны предоставить метод __eq__, если это возможно. Это обеспечивает остальные операторы сравнения, поэтому вам не нужно писать ни один из них самостоятельно.

Для полного набора функций сравнения я использовал следующий миксин, который вы можете поместить, например, в свой модуль, например, в файл mixin.py.

class ComparableMixin(object):
    def _compare(self, other, method):
        try:
            return method(self._cmpkey(), other._cmpkey())
        except (AttributeError, TypeError):
            # _cmpkey not implemented, or return different type,
            # so I can't compare with "other".
            return NotImplemented

    def __lt__(self, other):
        return self._compare(other, lambda s, o: s < o)

    def __le__(self, other):
        return self._compare(other, lambda s, o: s <= o)

    def __eq__(self, other):
        return self._compare(other, lambda s, o: s == o)

    def __ge__(self, other):
        return self._compare(other, lambda s, o: s >= o)

    def __gt__(self, other):
        return self._compare(other, lambda s, o: s > o)

    def __ne__(self, other):
        return self._compare(other, lambda s, o: s != o)

Чтобы использовать приведенный выше миксин, вам нужно реализовать метод _cmpkey(), который возвращает ключ объектов, которые можно сравнить, аналогично функции key(), используемой при сортировке. Реализация может выглядеть так:

>>> from .mixin import ComparableMixin

>>> class Orderable(ComparableMixin):
...
...     def __init__(self, firstname, lastname):
...         self.first = firstname
...         self.last = lastname
...
...     def _cmpkey(self):
...         return (self.last, self.first)
...
...     def __repr__(self):
...         return "%s %s" % (self.first, self.last)
...
>>> sorted([Orderable('Donald', 'Duck'), 
...         Orderable('Paul', 'Anka')])
[Paul Anka, Donald Duck]

Причина, по которой я использую это вместо рецепта total_ordering, заключается в этой ошибке. Это исправлено в Python 3.4, но часто вам нужно поддерживать и более старые версии Python.

Не уверен, что это завершено, но вы хотели бы определить:

__eq__, __gt__, __ge__, __lt__, __le__

Как сказал agf, мне не хватает:

__ne__

Вы сказали, что пытаетесь сделать это:

max((f(obj), obj) for obj in obj_list)[1]

Вы должны просто сделать это:

max(f(obj) for obj in obj_list)

РЕДАКТИРОВАТЬ: Или, как сказал гнибблер: max(obj_list, key=f)

Но вы сказали gnibbler, что вам нужна ссылка на объект max. Я думаю, что это самое простое:

def max_obj(obj_list, max_fn):
    if not obj_list:
        return None

    obj_max = obj_list[0]
    f_max = max_fn(obj)

    for obj in obj_list[1:]:
        if max_fn(obj) > f_max:
            obj_max = obj
    return obj_max

obj = max_obj(obj_list)

Конечно, вы можете позволить ему вызвать исключение, а не возвращать ничего, если вы попытаетесь найти max_obj() пустого списка.

Я просто придумал действительно хакерский способ сделать это. Это в том же духе, что вы изначально пытались сделать. Он не требует добавления каких-либо функций к объекту класса; это работает для любого класса.

max(((f(obj), obj) for obj in obj_list), key=lambda x: x[0])[1]

Мне это действительно не нравится, поэтому вот что-то менее краткое, что делает то же самое:

def make_pair(f, obj):
    return (f(obj), obj)

def gen_pairs(f, obj_list):
    return (make_pair(f, obj) for obj in obj_list)

def item0(tup):
    return tup[0]

def max_obj(f, obj_list):
    pair = max(gen_pairs(f, obj_list), key=item0)
    return pair[1]

Или вы можете использовать эту однострочную строку, если obj_list всегда является индексируемым объектом, таким как список:

obj_list[max((f(obj), i) for i, obj in enumerate(obj_list))[1]]

Это имеет то преимущество, что если есть несколько объектов, таких, что f(obj) возвращает идентичное значение, вы знаете, какой из них вы получите: тот, у которого самый высокий индекс, то есть самый последний в списке. Если вам нужен самый ранний из списка, вы можете сделать это с помощью ключевой функции.