20 February 2012

Python: сортировка списков методом .sort() с ключом — простыми словами

Python
Поводом опубликовать пост стало то, что при детальном изучении списков (массивов) в Python я не смог найти в сети ни одного простого описания метода сортировки элементов с использованием ключа: list.sort(key=...).

Может быть, конечно, это мне так не повезло и я долго понимаю простые для всех вещи, однако я думаю, что приведенная ниже информация будет весьма полезна таким же начинающим питонистам, как и я сам.

Итак, что мы имеем. Предположим, у нас есть список, который мы бы хотели отсортировать — и состоит он из трех строк разной длины в определенной последовательности:

sortList = ['a', 'сс', 'bbb']

Сортировка элементов массива методом .sort() производится по умолчанию лексикографически — проще говоря, в алфавитном порядке, а также от меньшего значения к большему. Поэтому если мы выполним:

sortList.sort()

то получим на выходе:

>>> ['a', 'bbb', 'cc']

Однако метод .sort() позволяет нам изменять и принцип, и порядок сортировки.

Для изменения принципа сортировки используется ключевое слово key, которое стало доступным начиная с версии Python 2.4.

Предположим, нам хотелось бы отсортировать наш список двумя способами: 1. в алфавитном порядке; 2. по длине строки. Первый способ, впрочем, уже работает как сортировка по умолчанию, однако мы можем добиться таких же результатов и с помощью параметра key:

sortList = ['a', 'cc', 'bbb']

# Создаем "внешнюю" функцию, которая будет сортировать список в алфавитном порядке:
def sortByAlphabet(inputStr):
return inputStr[0] # Ключом является первый символ в каждой строке, сортируем по нему

# Вторая функция, сортирующая список по длине строки:
def sortByLength(inputStr):
return len(inputStr) # Ключом является длина каждой строки, сортируем по длине

print u'Исходный список: ', sortList # >>> ['a', 'cc', 'bbb']

sortList.sort(key=sortByAlphabet) # Каждый элемент массива передается в качестве параметра функции
print u'Отсортировано в алфавитном порядке: ', sortList # >>> ['a', 'bbb', 'cc']

sortList.sort(key=sortByLength) # Каждый элемент массива передается в качестве параметра функции
print u'Отсортировано по длине строки: ', sortList # >>> ['a', 'cc', 'bbb']

# Теперь отсортируем по длине строки, но в обратном порядке:
sortList.sort(key=sortByLength, reverse=True) # В обратном порядке
print u'Отсортировано по длине строки, в обратном порядке: ', sortList # >>> ['bbb', 'cc', 'a']


Обратите внимание, что метод .sort() производит действия с исходным списком, переставляя элементы внутри него самого, и НЕ возвращает отсортированную копию исходного списка. Для получения отсортированной копии нужно использовать метод sorted:

newList = sorted(sortList)

— либо такой же вариант, но с параметром key (аналогично описанному выше):

newList = sorted(sortList, key=sortByLength)

У метода .sorted() есть и другие параметры, но мне они показались не настолько запутанными для самостоятельного разбора.
Tags:python.sort
Hubs: Python
0
324.2k 89
Comments 32
Top of the last 24 hours