Effective Python 第14条：sort方法排序指定key

sort方法简单使用

内置的列表提供了sort方法，可以根据多项指标给list实例中的元素排序。默认情况下是按照自然升序排列元素。
简单使用如下：

a_list = [5, 9, 10, 1, 6, 20, 12, 8]
a_list.sort()
print(a_list)
# [1, 5, 6, 8, 9, 10, 12, 20]

凡是具备自然顺序的内置类型几乎都可以使用sort方法排序，例如字符串、浮点数等。

自定义对象排序

一般对象如何排序呢？
如下面这个类，创建一些对象在列表里：

class Person:
    def __init__(self, name, age, weight):
        self.name = name
        self.age = age
        self.weight = weight

    def __repr__(self):
        return f"('name':{self.name}, 'age':{self.age}, 'weight':{self.weight})"


persons = [
    Person("Tom", 10, 40),
    Person("Bob", 11, 45),
    Person("Marin", 10, 37),
    Person("Cris", 9, 36),
]

如果只是这样写，那么persons 无法使用sort方法排序，因为sort方法发现没有排序的特殊方法定义在Person类中。

persons.sort()
# Traceback (most recent call last):
#   File "D:/demo01.py", line 18, in <module>
#     persons.sort()
# TypeError: '<' not supported between instances of 'Person' and 'Person'

自定义的一般类，排序一般是针对某些属性的。如果我们需要按照对象不同的属性进行排序，可以把排序逻辑定义成函数，然后把函数传给sort方法的key参数。这个排序函数应该具有返回值，且是个可以比较的值（也就是具有自然顺序的值），然后sort方法会根据这个值给对象排序。

通常使用lambda表达式来定义这个函数。
如下：

persons.sort(key=lambda x: x.name)
print(repr(persons))
# [('name':Bob, 'age':11, 'weight':45),
#   ('name':Cris, 'age':9, 'weight':36), 
#   ('name':Marin, 'age':10, 'weight':37),
#   ('name':Tom, 'age':10, 'weight':40)]

还可以按照某个值倒序排序。
如下，按照年龄倒序排序：

persons.sort(key=lambda x: x.age, reverse=True)
print(repr(persons))
# [('name':Bob, 'age':11, 'weight':45),
#   ('name':Tom, 'age':10, 'weight':40), 
#   ('name':Marin, 'age':10, 'weight':37), 
#   ('name':Cris, 'age':9, 'weight':36)]

字典列表排序

如果对象是序列、元组、字典等，还可以返回某个元素来排序。如下，按照字典的值给字典排序：

dict_list = [
    {"a": 3},
    {"b": 6},
    {"c": 1}
]
dict_list.sort(key=lambda x: list(x.values())[0])
print(dict_list)
# [{'c': 1}, {'a': 3}, {'b': 6}]

字符串列表排序

对于字符串，还可以做一些基本变化再排序。
如下一个字符串列表，忽而略大小写再排序：

places = ["home", "work", "New York", "Paris"]
places.sort()
print(places)  # ['New York', 'Paris', 'home', 'work']
places.sort(key=lambda x: x.lower())
print(places)  # ['home', 'New York', 'Paris', 'work']

多标准排序

如果需要多个排序标准，如上面的Person对象，先按照name，再按照年龄排序，这时候就需要构建一个元组，利用元组来排序：

persons.sort(key=lambda x: (x.name, x.age))
print(persons)
# [('name':Bob, 'age':11, 'weight':45),
#   ('name':Cris, 'age':9, 'weight':36), 
#   ('name':Marin, 'age':10, 'weight':37),
#   ('name':Tom, 'age':10, 'weight':40)]

但是当我们多个排序标准有的倒序有的正序怎么办呢？

当其中一项是数字时，那么实现key函数时，利用一元减操作符让指标顺序变反。比如，先按照年龄倒序，再按照体重正序：

persons.sort(key=lambda x: (-x.age, x.weight), reverse=False)
print(persons)
#[('name':Bob, 'age':11, 'weight':45),
#   ('name':Marin, 'age':10, 'weight':37),
#   ('name':Tom, 'age':10, 'weight':40),
#   ('name':Cris, 'age':9, 'weight':36)]

如果都是str，则不支持一元减操作符。

还有一个方法就是，分开进行两轮排序，但是注意：最优先排序的放最后。
比如，优先按照年龄倒序排序，再按照体重倒序排序，就要先排体重，再排年龄。

persons.sort(key=lambda x: x.weight, reverse=True)
print(persons)
# [('name':Bob, 'age':11, 'weight':45),
#   ('name':Tom, 'age':10, 'weight':40),
#   ('name':Marin, 'age':10, 'weight':37),
#   ('name':Cris, 'age':9, 'weight':36)]

persons.sort(key=lambda x: x.age, reverse=True)
print(persons)
# [('name':Bob, 'age':11, 'weight':45),
#   ('name':Tom, 'age':10, 'weight':40),
#   ('name':Marin, 'age':10, 'weight':37),
#   ('name':Cris, 'age':9, 'weight':36)]

这样的话无论多少排序项都可以实现了，而且每个指标都可以按照各自的方向排序。