问题
你想排序类型相同的对象,但是他们不支持原生的比较操作。
解决方案
内置的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 f'User{self.user_id}'
def sort_notcompare():
users=[User(23),User(3),User(99)]
print(users) # [User(23),User(3),User(99)]
print(sorted(users,key=lambda u:u.user_id)) # # ->[User3, User23, User99]
sort_notcompare()
另一种方式是使用operator.itemgetter()来替代lambda函数:
from operator import attrgetter
users = [User(23), User(3), User(99)]
print(sorted(users,key=attrgetter('user_id'))) # ->[User3, User23, User99]
讨论
选择使用lambda函数或者attrgetter()可能取决于个人喜好。但是,attrgetter()函数通常会运行快一点。并且还能同时运行多个字段进行比较。这个跟operator.itemgetter()函数作用类似。例如,如果User实例还有一个first_name和last_name属性,那么可以向下面这样排序:
by_name=sorted(users,key=attrgetter('last_name','first_name'))
同样需要注意的是,这一节用到的技术同样适用于向min()和max()之类的函数。比如:
print(min(users,key=attrgetter('user_id'))) # ->User3
print(max(users,key=attrgetter('user_id')))# ->User99