python-进阶教程-使用对象属性进行排序

0.摘要

python中的类通常不支持比较操作，但是我们可以依据类的属性值进行排序。

本文主要介绍使用operator.attrgetter()方法获取类的属性值，从而完成排序。

 

1.operator.attrgetter()方法

from operator import attrgetter

class User:
    def __init__(self, user_id):
        self.user_id = user_id
    def __repr__(self):
        return 'User({})'.format(self.user_id)

# Example
users = [User(23), User(3), User(99)]
print(users)
#result:[User(23), User(3), User(99)]

# Sort it by user-id
print(sorted(users, key=attrgetter('user_id')))
#result:[User(3), User(23), User(99)]

借助于lambda表达式也可实现同样的功能，但operator.attrgetter()方法执行效率更高。