首页 > 代码库 > Python优先权队列出现TypeError: unorderable types 解决方法
Python优先权队列出现TypeError: unorderable types 解决方法
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())
Python优先权队列出现TypeError: unorderable types 解决方法
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。