Власне сортування списку Python

Я переробляв якийсь свій старий код і натрапив на це:

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

Код працює (і я написав його 3 роки тому!), Але я не можу знайти цю річ, задокументовану де-небудь у документах Python, і всі використовують sorted()для реалізації користувацького сортування. Хтось може пояснити, чому це працює?

Це документально тут .

Метод sort () бере необов’язкові аргументи для контролю порівнянь.

cmp визначає спеціальну функцію порівняння двох аргументів (елементів списку), яка повинна повертати від'ємне, нульове або додатне число залежно від того, чи вважається перший аргумент меншим, рівним або більшим за другий аргумент: cmp = лямбда x, y : cmp (x.lower (), y.lower ()). Значення за замовчуванням - None.

Як зауваження, ось краща альтернатива для здійснення того ж сортування:

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

Або як варіант:

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

Перегляньте сортування , як це дуже корисно.

Так само, як цей приклад. Ви хочете відсортувати цей список.

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

вихід:

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

вам слід відсортувати кортежі за другим елементом, а потім за першим:

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

Нарешті:

просто sort(letter_cmp)

У Python 3 це не працює.

Ви можете використовувати functools cmp_to_key, щоб функції порівняння в старому стилі працювали.

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)

У python3 ви можете написати:

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

Навіть краще:

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

Взято з: https://docs.python.org/3/howto/sorting.html