Python3 sort 自定义排序逻辑：lambda函数、可比较类、cmp_to_key

以字符串排序为例，现在我们有这样一个字符串的 list ：

words = ["vanilla", "tom", "sue", "david", "bill", "tom", "jezebel", "jackb", "jack", "jacka"]

默认排序方式

对于字符串而言，其默认的比较大小逻辑为：按首个字符顺序比较大小，相同则看后一个字符，后一个字符为空者较小。

def test1(things):
    things.sort()
    print(things)
    
test1(words.copy())

输出：

['bill', 'david', 'jack', 'jacka', 'jackb', 'jezebel', 'sue', 'tom', 'tom', 'vanilla']

lambda函数

sort 方法可以接收 key 参数，传递一个如下的可调用的对象：
以下例子通过 lambda 函数设置新的排序逻辑：以字符串长度为基准。
lambda 函数需要：1.接收1个参数；2.返回一个值，且返回值是可比较大小的。

def test2(things):
    things.sort(key=lambda e: len(e))
    print(things)

test2(words.copy())

输出：

['tom', 'sue', 'tom', 'bill', 'jack', 'david', 'jackb', 'jacka', 'vanilla', 'jezebel']

自定义可比较类

但是 lambda 函数无法表示比较复杂的排序逻辑。比如我们想：优先按照字符串长度为基准排序，再按照字母大小排序。lambda 函数是做不到如此复杂的逻辑的。
实际上， key 参数也可以传接收1个参数以初始化的类，比如下面的继承 str 类的 Cmp 类（因为 str 本身就接收一个参数以初始化，所以无需重写 Cmp 的 __init__ 函数）
为了使 Cmp 是可比较大小的，需定义用于比较的 magic method，这些函数为: __gt__, __lt__,__ge__, __le__, __eq__，分别在进行 >, <, >=, <=, == 运算时被调用，一个鲁棒的可比较类需要包含全部这五个函数的定义。
Cmp 中只重写了 __lt__ ，因为在 sort 的执行中，优先进行 < 运算（只有在 < 运算抛出异常的情况下，才会试图进行其它比较运算）

def test3(things):
    class Cmp(str):
        def __lt__(self, other):
            if len(self) == len(other):
                return str.__lt__(self, other)
            else:
                return len(self) < len(other)
    things.sort(key=Cmp)
    print(things)

test3(words.copy())

输出：

['sue', 'tom', 'tom', 'bill', 'jack', 'david', 'jacka', 'jackb', 'jezebel', 'vanilla']

通过cmp_to_key自定义比较逻辑

定义鲁棒的 Cmp 类比较麻烦，毕竟两个对象的大小比较，无外乎“大于”、“小于”、“等于”三个状态。最直接的方法难道不是定义一个函数，输入两个参数，输出前者与后者之间的大小关系吗？可是有这样的接口吗？有的。
通过 functools 模块里的 cmp_to_key 函数，可以简洁地将上述自定义函数转化成 key 可接收的格式。
自定义函数需要：1.接收两个参数 p1, p2；2.返回1、0或-1，其中1代表 p1 > p2，0代表 p1 == p2， -1代表 p1 < p2。

def test4(things):
    def compare(s1, s2):
        if len(s1) == len(s2):
            for c1, c2 in zip(s1, s2):
                if c1 > c2:
                    return 1
                elif c1 < c2:
                    return -1
            return 0  # 这里的比较其实可以直接使用字符串之间的 >, <, == 实现，之所以手动实现，是为了强调，没有内置比较函数的类也可以手动实现自定义的比较
        else:
            if len(s1) > len(s2):
                return 1
            else:
                return -1
    from functools import cmp_to_key
    things.sort(key=cmp_to_key(compare))
    print(things)

test4(words.copy())

输出：

['sue', 'tom', 'tom', 'bill', 'jack', 'david', 'jacka', 'jackb', 'jezebel', 'vanilla']