Python3 - 排序不支持原生比较的对象

问题

你想排序类型相同的对象，但是他们不支持原生的比较操作。

解决方案

内置的 sorted() 函数有一个关键字参数 key ，可以传入一个 callable 对象给它， 这个 callable 对象对每个传入的对象返回一个值，这个值会被 sorted 用来排序这些对象。

比如，如果你在应用程序里面有一个 User 实例序列，并且你希望通过他们的 user_id 属性进行排序， 你可以提供一个以 User 实例作为输入并输出对应 user_id 值的 callable 对象。比如：

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

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

def sort_class():
    users = [User(23), User(3), User(99)]
    users_sort = sorted(users, key=lambda x: x.user_id)
    print(users)
    return users_sort

print(sort_class())

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

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

users = [User(23), User(3), User(99)]
users_sort = sorted(users, key=attrgetter('user_id'))
print(users_sort)

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

讨论

选择使用 lambda 函数或者是 attrgetter() 取决于个人喜好。 但是， attrgetter() 函数通常会运行的快点，并且还能同时允许多个字段进行比较。 这个跟 operator.itemgetter() 函数作用于字典类型类似。

本节用到的方法，同样适用于像 min() 和 max() 之类的函数。比如：

users_min = min(users, key=attrgetter('user_id'))
print(users_min)

User(3)