Python优先权队列出现TypeError: unorderable types 解决方法

£可爱£侵袭症+ 提交于 2020-11-24 06:26:05
class MyType:
    def __init__(self, arg1, arg2=None):
        self.data1 = arg1
        self.data2 = arg2
q = queue.PriorityQueue()
m1 = MyType(5)
m2 = MyType(2)
m3 = MyType(4)

q.put( (1, m1))
q.put( (2, m2))
q.put( (3, m3))

while not q.empty():
    print( q.get())

        PriorityQueue 里的数据格式是 (优先级, 数据)  其中优先级为一个整形数字, 数据可以是内置类型或者自定义类型,当队列里的所有数据项的优先级没有重复时上面的代码没有什么问题,但万一两个数据项的优先级相同就会报错,类似这样:

        因为python的优先级队列使用了堆排序,当队列中两个数据项优先级相同时,队列就会把两个数据项直接拿来再比较确定数据的排列顺序,但是由于我们使用了自定义的数据类型,系统不知道如何比较我们定义的数据,所以就会报错。

        所以为了让我们自定义的数据可以比较大小,我们可以覆盖 __lt__这个函数,实现小于比较操作。然后python报错就解决了。(觉得这玩意和c++的stl一样 , stl中也需要类重载<运算符)

class MyType:
    def __init__(self, arg1, arg2=None):
        self.data1 = arg1
        self.data2 = arg2
    def __lt__(self, y):    #实现<操作
        return self.data1<y.data1
q = queue.PriorityQueue()
m1 = MyType(5)
m2 = MyType(2)
m3 = MyType(4)

q.put( (2, m1))
q.put( (2, m2))
q.put( (3, m3))

while not q.empty():
    print( q.get())


易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!