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

在编程中经常会遇到自定义对象的排序或者堆维护的问题,这就需要数据结构中的元素可比较大小,由于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]
发布了24 篇原创文章 · 获赞 8 · 访问量 7222

猜你喜欢

转载自blog.csdn.net/geek_hch/article/details/88614698