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

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

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]

你可能感兴趣的:(学习)