Python: attrgetter()函数: 排序不支持原生比较的对象

想排序类型相同的对象,但是他们不支持原生的比较操作。

  • 内置的sorted()函数有一个关键字参数key,可传入一个callable对象给它

  • 这个callabel对象对每个传入的对象返回一个值,这个值被sorted()用来排序这些对象

1.lambda 函数

有一个User 实例序列,希望通过他们的user id 属性进行排序,可提供一个以User 实例作为输入并输出对应user id 值的callable 对象

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


def sort_notcompare():
	users = [User(23), User(3), User(99)]
	print(users)
	print(sorted(users, key=lambda u: u.user_id))

2.operator.attrgetter()

另外一种方式是使用operator.attrgetter() 来代替lambda 函数:

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:778463939
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
>>> from operator import attrgetter
>>> sorted(users, key=attrgetter('user_id'))
[User(3), User(23), User(99)]

attrgetter() 函数通常会运行的快点,并且还能同时允许多个字段进行比较。

如果User 实例还有一个first name 和last name 属性排序,

>>>by_name = sorted(users, key=attrgetter('last_name', 'first_name'))

3.min() 和max()

适用于像min() 和max() 之类的函数。

>>> min(users, key=attrgetter('user_id')
User(3)
>>> max(users, key=attrgetter('user_id')
User(99)
posted @ 2022-10-07 20:30  I'm_江河湖海  阅读(1)  评论(0编辑  收藏  举报