Python自定义类的大小比较及排序

最新推荐文章于 2023-08-07 10:39:09 发布

三七、

最新推荐文章于 2023-08-07 10:39:09 发布

阅读量1w

点赞数 5

分类专栏：编程语言文章标签： python

本文链接：https://blog.youkuaiyun.com/geek_hch/article/details/88614698

版权

编程语言专栏收录该内容

7 篇文章

订阅专栏

在编程中经常会遇到自定义对象的排序或者堆维护的问题，这就需要数据结构中的元素可比较大小，由于heapq或者list.sort()都只使用小于<比较，所以只需要定义类的__lt__()方法即可，类似c++中的运算符重载，在Python3中，所有类都有该默认函数，其他函数见下表
在这里插入图片描述　　此处我们只需要重写该函数, 然后就可以愉快的进行排序比较啦：

class OB():
    def __init__(self,num):
        self.num = num
    def __lt__(self,other):
        return self.num<other.num

l = [OB(9),OB(4),OB(7),OB(1),OB(7)]
l.sort()
print([e.num for e in l])

#输出结果： [1, 4, 7, 7, 9]