Jak posortować listę obiektów na podstawie atrybutu obiektów?

Mam listę obiektów Python, które chciałbym posortować według atrybutu samych obiektów. Lista wygląda następująco:

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

Każdy obiekt ma liczbę:

>>> ut[1].count
1L

Muszę posortować listę według liczby malejących.

Widziałem kilka metod, ale szukam najlepszych praktyk w Pythonie.

# 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)

Więcej informacji na temat sortowania według kluczy .

Najlepszym sposobem, który może być najszybszy, szczególnie jeśli twoja lista zawiera wiele rekordów, jest użycie operator.attrgetter("count"). Może to jednak działać na wcześniejszej wersji Pythona, więc dobrze byłoby mieć mechanizm awaryjny. Następnie możesz wykonać następujące czynności:

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

Czytelnicy powinni zauważyć, że metoda key =:

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

jest wiele razy szybsza niż dodawanie do obiektów bogatych operatorów porównania. Byłem zaskoczony, gdy to przeczytałem (str. 485 „Python w pigułce”). Możesz to potwierdzić, uruchamiając testy w tym małym programie:

#!/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

Moje, bardzo minimalne testy pokazują, że pierwsze sortowanie jest ponad 10 razy wolniejsze, ale książka mówi, że ogólnie jest tylko około 5 razy wolniejsze. Mówią, że powodem jest wysoce zoptymalizowany algorytm sortowania używany w pythonie ( timsort ).

Jednak bardzo dziwne jest to, że .sort (lambda) jest szybszy niż zwykły stary .sort (). Mam nadzieję, że to naprawią.

Podejście obiektowe

Dobrą praktyką jest, aby logika sortowania obiektów, jeśli ma zastosowanie, była właściwością klasy, a nie włączana w każdym przypadku, w którym wymagane jest porządkowanie.

Zapewnia to spójność i eliminuje potrzebę stosowania kodu płyty kotłowej.

Co najmniej należy określić __eq__i __lt__działania, aby to zadziałało. Więc po prostu użyj 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)

Wygląda bardzo podobnie do listy instancji modelu Django ORM.

Dlaczego nie posortować ich według zapytania:

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

Dodaj operatory porównania bogatego do klasy obiektowej, a następnie użyj metody sort () z listy.
Zobacz bogate porównanie w pythonie .

Aktualizacja : Chociaż ta metoda zadziałałaby, myślę, że rozwiązanie z Tryptyku lepiej pasuje do twojego przypadku, ponieważ jest prostsze.

Jeśli atrybut, który chcesz posortować, jest właściwością , możesz uniknąć importowania operator.attrgetteri fgetzamiast tego użyć metody właściwości .

Na przykład dla klasy Circlez właściwością radiusmożemy posortować listę circleswedług promieni w następujący sposób:

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

To nie jest najbardziej znana funkcja, ale często zapisuje mi linię przy imporcie.