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()方法执行效率更高。