python cookbook 学习笔记 第一章 数据结构和算法(13)排序不支持原生比较的对象

• 问题
  • 想要排序类型相同的对象，但他们不支持原生的比较操作。
• 解决方案
  • 内置的函数 sorted() 函数有一个关键字参数key,可以传入一个callable对象给它，这个callable对象 对每个传入的对象返回一个值，这个值会被sorted用来排序这些对象。比如，如果在应用程序里面有一个User 实例序列，并且希望通过他们的User_id属性来进行排序，可以提高一个以User实例作为输入并输出对应User_id 值的 callable 对象。比如：

class User():
    def __init__(self, user_id):
        self.user_id = user_id

    def __repr__(self):
        return "User({})".format(self.user_id)

def sort_notcompare():
    users = [User(23),User(3),User(99)]
    print(users)
    print(sorted(users,key= lambda u: u.user_id))  

sort_notcompare()  # [User(3), User(23), User(99)]

• 另外一种方式是使用 operator.attrgetter() 来代替 lambda 函数：

from operator import attrgetter

sorted(users,key= attrgetter("user_id"))

• 讨论： 选择使用 lambda 函数或者是 attrgetter() 可能取决于个人喜好。但是 attrgetter() 函数 通常会运行快点，并且还能同时允许多个字段进行比较。这个跟operator.itemgetter()函数作用于字典类 型很类似。例如，如果 User 实例还有一个first_name 和 last_name 属性，那么可以像下面这样排序：

sorted(users, key= attrgetter("last_name","first_name"))

• 同时需要注意的是，这一小节讲到的演示，同样使用于像min()和max()之类的函数。比如：

min(users, key= attrgetter("user_id"))  # User(3)
max(users, key= attrgetter("user_id"))  # User(99)