Làm thế nào để sắp xếp danh sách các đối tượng dựa trên một thuộc tính của các đối tượng?

Tôi đã có một danh sách các đối tượng Python mà tôi muốn sắp xếp theo một thuộc tính của chính các đối tượng đó. Danh sách này trông như:

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

Mỗi đối tượng có một số lượng:

>>> ut[1].count
1L

Tôi cần sắp xếp danh sách theo số lượng giảm dần.

Tôi đã thấy một số phương pháp cho việc này, nhưng tôi đang tìm cách thực hành tốt nhất trong 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)

Thêm về cách sắp xếp theo các phím .

Một cách có thể nhanh nhất, đặc biệt nếu danh sách của bạn có nhiều hồ sơ, là sử dụng operator.attrgetter("count"). Tuy nhiên, điều này có thể chạy trên phiên bản tiền vận hành của Python, vì vậy thật tuyệt khi có cơ chế dự phòng. Bạn có thể muốn làm như sau, sau đó:

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

Bạn đọc nên chú ý rằng key = phương thức:

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

nhanh hơn nhiều lần so với việc thêm các toán tử so sánh phong phú vào các đối tượng. Tôi đã rất ngạc nhiên khi đọc nó (trang 485 của "Python in a Nutshell"). Bạn có thể xác nhận điều này bằng cách chạy thử nghiệm trên chương trình nhỏ này:

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

Các thử nghiệm của tôi, rất tối thiểu, cho thấy loại đầu tiên chậm hơn 10 lần, nhưng cuốn sách nói rằng nó chỉ chậm hơn khoảng 5 lần nói chung. Lý do họ nói là do thuật toán sắp xếp tối ưu hóa cao được sử dụng trong python ( timsort ).

Tuy nhiên, rất kỳ lạ là .sort (lambda) nhanh hơn so với .sort () cũ. Tôi hy vọng họ khắc phục điều đó.

Phương pháp hướng đối tượng

Đó là cách thực hành tốt để tạo logic sắp xếp đối tượng, nếu có thể, một thuộc tính của lớp thay vì được kết hợp trong mỗi trường hợp, việc đặt hàng là bắt buộc.

Điều này đảm bảo tính nhất quán và loại bỏ sự cần thiết của mã soạn sẵn.

Tối thiểu, bạn nên chỉ định __eq__và __lt__hoạt động để làm việc này. Sau đó chỉ cần sử dụng 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)

Nó trông giống như một danh sách các trường hợp mô hình Django ORM.

Tại sao không sắp xếp chúng theo truy vấn như thế này:

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

Thêm các toán tử so sánh phong phú vào lớp đối tượng, sau đó sử dụng phương thức sort () của danh sách.
Xem so sánh phong phú trong trăn .

Cập nhật : Mặc dù phương pháp này sẽ hoạt động, tôi nghĩ giải pháp từ Triptych phù hợp hơn với trường hợp của bạn vì cách đơn giản hơn.

Nếu thuộc tính bạn muốn sắp xếp theo là một thuộc tính , thì bạn có thể tránh nhập operator.attrgettervà sử dụng thuộc tínhfget phương thức thay thế.

Ví dụ, đối với một lớp Circlecó thuộc tính, radiuschúng ta có thể sắp xếp danh sách circlestheo bán kính như sau:

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

Đây không phải là tính năng nổi tiếng nhất nhưng thường giúp tôi tiết kiệm một dòng với việc nhập.