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

StackOverflow https://stackoverflow.com/questions/403421

  •  03-07-2019
  •  | 
  •  

Вопрос

У меня есть список объектов 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

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

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

во много раз быстрее, чем добавление к объектам расширенных операторов сравнения.Я был удивлен, прочитав это (стр. 485 книги «Коротко о Python»).Вы можете убедиться в этом, запустив тесты на этой маленькой программе:

#!/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 (Тимсорт).

Тем не менее, очень странно, что .sort(lambda) работает быстрее, чем старый добрый .sort().Я надеюсь, что они это исправят.

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

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

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

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

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

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

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

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

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


Обновлять:Хотя этот метод сработает, я думаю, что решение из «Триптиха» лучше подойдет для вашего случая, потому что оно более простое.

Лицензировано под: CC-BY-SA с атрибуция
Не связан с StackOverflow
scroll top