it-swarm.com.ru

Как отсортировать список объектов на основе атрибута объектов?

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

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

Каждый объект имеет количество:

>>> ut[1].count
1L

Мне нужно отсортировать список по убыванию количества отсчетов.

Я видел несколько методов для этого, но я ищу лучшую практику в Python.

653
Nick Sergeant
# 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)

Подробнее о сортировка по ключам "

1100
Triptych

Самый быстрый способ, особенно если в вашем списке много записей, это использовать 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
71
tzot

Читатели должны заметить, что ключ = метод:

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 (). Я надеюсь, что они исправят это.

57
Jose M Vidal
from operator import attrgetter
ut.sort(key = attrgetter('count'), reverse = True)
33
attrgetter

Объектно-ориентированный подход

Хорошей практикой является сделать логику сортировки объектов, если это применимо, свойством класса, а не включать его в каждом случае, когда требуется упорядочение.

Это обеспечивает согласованность и устраняет необходимость в шаблонном коде.

Как минимум, вы должны указать операции __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]
30
jpp

Это очень похоже на список Django экземпляров модели ORM.

Почему бы не отсортировать их по запросу так:

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

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


Обновление : хотя этот метод будет работать, я думаю, что решение от Triptych лучше подходит для вашего случая, потому что это намного проще.

10
rob