Pythonソートチートシート

[ソート HOW TO — Python 3.5.1 ドキュメント] (http://docs.python.jp/3/howto/sorting.html) の内容と、追加のTIPSです。

ソート関数

list.sort()

• リストをソートして元のリストを置き換える。
• 戻り値はない。

>>> a = [3, 0, 5, 1]
>>> a.sort()
>>> a
[0, 1, 3, 5]
>>> print(a.sort())  # 戻り値はない
None

sorted(list)

• リストをソートした結果を返す。
• 元のリストは変更されない。
• リスト以外にも使える。

>>> a = [3, 0, 5, 1]
>>> sorted(a)
[0, 1, 3, 5]
>>> a  # 元のリストは変更されない
[3, 0, 5, 1]
>>> sorted(iter([3, 0, 5, 1]))  # リスト以外にも使える
[0, 1, 3, 5]

逆順ソート

• reverse=Trueを指定する。

>>> a = [3, 0, 5, 1]
>>> sorted(a, reverse=True)
[5, 3, 1, 0]

ソートキーを指定する

• keyに関数やラムダ式を指定する。

>>> a = ["c", "aaa", "Bb"]
>>> sorted(a, key=lambda x: x.lower())
['aaa', 'Bb', 'c']
>>> sorted(a, key=str.lower)
['aaa', 'Bb', 'c']
>>> sorted(a, key=len)
['c', 'Bb', 'aaa']

辞書・リスト・タプルを要素に持つリストのソート

• デフォルトでは要素を順に比較してくれる。

>>> a = [("b", 3), ("b", 1), ("a", 1), ("a", 2), ("c", 2)]
>>> sorted(a)
[('a', 1), ('a', 2), ('b', 1), ('b', 3), ('c', 2)]

• キーを指定する場合はitemgetterを使用するとラムダ式より簡潔。

>>> a = [{"key1": 3}, {"key1": 1}]
>>> from operator import itemgetter
>>> sorted(a, key=itemgetter("key1"))
[{'key1': 1}, {'key1': 3}]

クラスを要素に持つリストのソート

• キーを指定する場合はattrgetterを使用するとラムダ式より簡潔。

>>> from datetime import date
>>> a = [date(2016, 1, 1), date(2015, 12, 1)]
>>> from operator import attrgetter
>>> sorted(a, key=attrgetter("month"))
[datetime.date(2016, 1, 1), datetime.date(2015, 12, 1)]

複合ソート

• ソート結果に対して更にソートをすれば複合ソートができる。
• 優先度の低いキーから先にソートしていく。

>>> a = [("b", 3), ("b", 1), ("a", 1), ("a", 2), ("c", 2)]
>>> b = sorted(a, key=itemgetter(1), reverse=True)
>>> sorted(b, key=itemgetter(0))
[('a', 2), ('a', 1), ('b', 3), ('b', 1), ('c', 2)]

• itemgetterとattrgetterは一度に複数キーを指定できるので、降順・昇順が全てのキーで同じならこちらを使う。

>>> sorted(a, key=itemgetter(0, 1))
[('a', 1), ('a', 2), ('b', 1), ('b', 3), ('c', 2)]

Noneを含む場合のソート

• 普通にソートすると失敗する。
• 複合ソートとして第一キーにNoneかどうかのbool、第二キーに要素自身を指定する。

>>> a = [3, 0, None, 5, 1]
>>> sorted(a)  # 普通にソートすると失敗
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: unorderable types: NoneType() < int()
>>> sorted(a, key=lambda x: (x is None, x))  # Noneを末尾に
[0, 1, 3, 5, None]
>>> sorted(a, key=lambda x: (x is not None, x))  # Noneを先頭に
[None, 0, 1, 3, 5]

Python sort list with None at the end - Stack Overflow

クラスをソート可能にする

• __lt__()を実装しておく。

class P(object):
    def __init__(self, gender, age):
        self.gender = gender
        self.age = age
    def __repr__(self):
        return "{:s}{:d}".format(self.gender, self.age)
    def __lt__(self, other):
        return (self.gender, self.age) < (other.gender, other.age)

a = [P("F", 30), P("F", 20), P("M", 20), P("M", 40), P("F", 10)]
print(sorted(a))

複数のソートされたiterablesをマージする

• heapq.mergeを使用する。

>>> import heapq
>>> list(heapq.merge([1, 3, 4, 7], [2, 5], [6]))
[1, 2, 3, 4, 5, 6, 7]

巨大なテキストファイルをソートする

• Pythonででっかいテキストファイルをソートする - Qiita