Sắp xếp danh sách Python tùy chỉnh

Tôi đang cấu trúc lại một số mã cũ của mình và bắt gặp điều này:

alist.sort(cmp_items)

def cmp_items(a, b):
    if a.foo > b.foo:
        return 1
    elif a.foo == b.foo:
        return 0
    else:
        return -1

Mã hoạt động (và tôi đã viết nó cách đây 3 năm!) Nhưng tôi không thể tìm thấy điều này được ghi lại ở bất kỳ đâu trong tài liệu Python và mọi người đều sử dụng sorted()để triển khai sắp xếp tùy chỉnh. Ai đó có thể giải thích tại sao điều này hoạt động?

Nó được ghi lại ở đây .

Phương thức sort () nhận các đối số tùy chọn để kiểm soát các so sánh.

cmp chỉ định một hàm so sánh tùy chỉnh của hai đối số (mục danh sách) sẽ trả về số âm, số 0 hoặc số dương tùy thuộc vào việc đối số đầu tiên được coi là nhỏ hơn, bằng hoặc lớn hơn đối số thứ hai: cmp = lambda x, y : cmp (x.lower (), y.lower ()). Giá trị mặc định là không có.

Lưu ý thêm, đây là một giải pháp thay thế tốt hơn để thực hiện cùng một phân loại:

alist.sort(key=lambda x: x.foo)

Hay cách khác:

import operator
alist.sort(key=operator.attrgetter('foo'))

Kiểm tra Cách sắp xếp , nó rất hữu ích.

Cũng giống như ví dụ này. Bạn muốn sắp xếp danh sách này.

[('c', 2), ('b', 2), ('a', 3)]

đầu ra:

[('a', 3), ('b', 2), ('c', 2)]

bạn nên sắp xếp các bộ theo mục thứ hai, sau đó đến mục đầu tiên:

def letter_cmp(a, b):
    if a[1] > b[1]:
        return -1
    elif a[1] == b[1]:
        if a[0] > b[0]:
            return 1
        else:
            return -1
    else:
        return 1

Cuối cùng:

chỉ sort(letter_cmp)

Điều này không hoạt động trong Python 3.

Bạn có thể sử dụng functools cmp_to_key để các hàm so sánh kiểu cũ hoạt động.

from functools import cmp_to_key

def cmp_items(a, b):
    if a.foo > b.foo:
        return 1
    elif a.foo == b.foo:
        return 0
    else:
        return -1

cmp_items_py3 = cmp_to_key(cmp_items)

alist.sort(cmp_items_py3)

Trong python3, bạn có thể viết:

from functools import cmp_to_key

def cmp_items(a, b):
    if a.foo > b.foo:
        return 1
    elif a.foo == b.foo:
        return 0
    else:
        return -1

alist = sorted(alist, key=cmp_to_key(cmp_items))

Thậm chí còn tốt hơn:

student_tuples = [
    ('john', 'A', 15),
    ('jane', 'B', 12),
    ('dave', 'B', 10),
]

sorted(student_tuples, key=lambda student: student[2])   # sort by age
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]

Lấy từ: https://docs.python.org/3/howto/sorting.html