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