排序没有本地比较支持的对象
假设有一组相同类的对象,类未实现本地比较。现在想根据类的某个属性进行排序。
常规的做法是用sorted方法的关键字参数key进行定制化比较,如下示例:
class User:
def __init__(self, user_id):
self.user_id = user_id
def __repr__(self):
return 'User({})'.format(self.user_id)
>>> users = [User(23), User(3), User(99)]
>>> users
[User(23), User(3), User(99)]
>>> sorted(users, key=lambda u: u.user_id)
[User(3), User(23), User(99)]
>>>
一个替代方案是可以使用operator模块的attrgetter函数,如下:
>>> from operator import attrgetter
>>> sorted(users, key=attrgetter('user_id'))
[User(3), User(23), User(99)]
使用常规的lambda还是attrgetter依据个人偏好,但attrgetter在性能上更胜一筹,且增加了一个属性,可以同时支持多个类属性域。。attrgetter类似上一博文itemgetter的用法。