排序没有本地比较支持的对象

假设有一组相同类的对象,类未实现本地比较。现在想根据类的某个属性进行排序。

常规的做法是用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的用法。

posted @ 2019-08-06 16:37  Jeffrey_Yang  阅读(140)  评论(0编辑  收藏  举报