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())
来源:oschina
链接:https://my.oschina.net/u/1174481/blog/306991