- 问题
- 想要排序类型相同的对象,但他们不支持原生的比较操作。
- 解决方案
- 内置的函数 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)