Як сортувати список об’єктів на основі атрибутів об’єктів?

У мене є список об’єктів Python, які я хотів би сортувати за атрибутом самих об’єктів. Список виглядає так:

>>> ut
[<Tag: 128>, <Tag: 2008>, <Tag: <>, <Tag: actionscript>, <Tag: addresses>,
 <Tag: aes>, <Tag: ajax> ...]

Кожен об’єкт має кількість:

>>> ut[1].count
1L

Мені потрібно сортувати список за кількістю підрахунків, що спадають.

Я бачив кілька методів цього, але шукаю кращі практики в Python.

# To sort the list in place...
ut.sort(key=lambda x: x.count, reverse=True)

# To return a new list, use the sorted() built-in function...
newlist = sorted(ut, key=lambda x: x.count, reverse=True)

Детальніше про сортування за клавішами .

Швидкий спосіб, особливо якщо у вашому списку багато записів, є використання operator.attrgetter("count"). Однак це може працювати на дооператорській версії Python, тому непогано було б створити резервний механізм. Ви можете зробити наступне:

try: import operator
except ImportError: keyfun= lambda x: x.count # use a lambda if no operator module
else: keyfun= operator.attrgetter("count") # use operator since it's faster than lambda

ut.sort(key=keyfun, reverse=True) # sort in-place

Читачі повинні помітити, що ключ = метод:

ut.sort(key=lambda x: x.count, reverse=True)

у багато разів швидше, ніж додавання об'єктів багатих операторів порівняння. Я був здивований, прочитавши це (стор. 485 «Пітона в горішці»). Ви можете підтвердити це, запустивши тести на цій маленькій програмі:

#!/usr/bin/env python
import random

class C:
    def __init__(self,count):
        self.count = count

    def __cmp__(self,other):
        return cmp(self.count,other.count)

longList = [C(random.random()) for i in xrange(1000000)] #about 6.1 secs
longList2 = longList[:]

longList.sort() #about 52 - 6.1 = 46 secs
longList2.sort(key = lambda c: c.count) #about 9 - 6.1 = 3 secs

Мої, дуже мінімальні тести, показують, що перший сорт більш ніж у 10 разів повільніше, але в книзі йдеться про те, що загалом це лише приблизно в 5 разів. Причина, по якій вони кажуть, пов'язана з високооптимізованим алгоритмом сортування, який використовується в python ( timsort ).

І все-таки дуже дивно, що .sort (лямбда) швидше звичайного старого .sort (). Я сподіваюся, що вони це виправлять.

Об'єктно-орієнтований підхід

Доброю практикою є створення логіки сортування об'єктів, якщо це застосовно, властивістю класу, а не включеним у кожному випадку, впорядкування потрібно.

Це забезпечує узгодженість та позбавляє від необхідності кодового коду.

Як мінімум, слід вказати __eq__і __lt__операції, щоб це працювало. Тоді просто використовуйте sorted(list_of_objects).

class Card(object):

    def __init__(self, rank, suit):
        self.rank = rank
        self.suit = suit

    def __eq__(self, other):
        return self.rank == other.rank and self.suit == other.suit

    def __lt__(self, other):
        return self.rank < other.rank

hand = [Card(10, 'H'), Card(2, 'h'), Card(12, 'h'), Card(13, 'h'), Card(14, 'h')]
hand_order = [c.rank for c in hand]  # [10, 2, 12, 13, 14]

hand_sorted = sorted(hand)
hand_sorted_order = [c.rank for c in hand_sorted]  # [2, 10, 12, 13, 14]

from operator import attrgetter
ut.sort(key = attrgetter('count'), reverse = True)

Це схоже на список екземплярів моделі Django ORM.

Чому б не сортувати їх за таким запитом:

ut = Tag.objects.order_by('-count')

Додайте розширені оператори порівняння до класу об’єктів, а потім використовуйте метод sort () списку.
Дивіться багате порівняння в python .

Оновлення : Хоча цей метод спрацював би, я думаю, що рішення від триптиху краще підходить для вашого випадку, оскільки спосіб простіший.

Якщо атрибут, який ви хочете сортувати, - це властивість , то ви можете уникнути імпорту operator.attrgetterта використовувати fgetнатомість метод властивості .

Наприклад, для класу Circleз властивістю radiusми можемо сортувати список circlesза радіусами наступним чином:

result = sorted(circles, key=Circle.radius.fget)

Це не найвідоміша функція, але часто економить мені рядок із імпортом.