python数据结构算法题十七:排序不支持原生比较的对象

问题:

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

解决方案:

内置的 sorted() 函数有一个关键字参数 key ,可以传入一个 callable 对象给 它,这个 callable 对象对每个传入的对象返回一个值,这个值会被 sorted 用来排序 这些对象。

比如,如果你在应用程序里面有一个 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))

使用 operator.attrgetter() 来代替 lambda 函数:


>>> from operator import attrgetter
>>> sorted(users, key=attrgetter('user_id')) [User(3), User(23), User(99)]
>>>

选择使用 lambda 函数或者是 attrgetter() 可能取决于个人喜好。但是, attrgetter() 函数通常会运行的快点,并且还能同时允许多个字段进行比较。这 个跟 operator.itemgetter() 函数作用于字典类型很类似。

例如, 如果 User 实例还有一个 first_name 和 last_name 属性,那么可以向下面这样排序。

>>> min(users, key=attrgetter('user_id')) User(3)
>>> max(users, key=attrgetter('user_id')) User(99)
>>>

猜你喜欢

转载自blog.csdn.net/m0_68635815/article/details/135442306