VB.net 2010 视频教程 VB.net 2010 视频教程 python基础视频教程
SQL Server 2008 视频教程 c#入门经典教程 Visual Basic从门到精通视频教程
当前位置:
首页 > 编程开发 > python入门教程 >
  • 1.14 排序不支持原生比较的对象

问题描述

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

解决方案

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

比如有一个User实例序列,你希望通过它们的user_id属性排序,那么你可以提供一个以User实例为输入,对应的user_id为输出的callable对象。比如如下的lambda函数:

class User:
    def __init__(self, user_id):
        self.user_id = user_id

    def __repr__(self):
        return 'User({})'.format(self.user_id)


def sort_users():
    users = [User(23), User(3), User(99)]
    print(users)
    print(sorted(users, key=lambda u: u.user_id))
	
"""sort_users()输出结果:
[User(23), User(3), User(99)]
[User(3), User(23), User(99)]
"""

也可以用operator.attrgetter()替代lambda函数:

from operator import attrgetter
sorted(users, key=attrgetter('user_id'))

讨论

通常attrgetter()函数运行的更快一些,并且可以支持多字段比较。这个跟operator.itemgetter()函数作用于字典类型很类似(参考1.13小节)。例如,如果User实例还有一个name属性,那么可以像这样排序:sorted(users, key=attrgetter('user_id', 'name'))

同样的,这一方法也适用于像min()max()之类的函数。比如:

min(users, key=attrgetter('user_id'))  # User(3)
max(users, key=attrgetter('user_id'))  # User(99)
 
原文:https://www.cnblogs.com/L999C/p/15745619.html


相关教程