python中PriorityQueue的理解

babybin

已于 2024-01-21 18:32:04 修改

阅读量1.1w

点赞数 6

分类专栏： Python 文章标签： python

于 2020-12-24 14:24:31 首次发布

本文链接：https://blog.csdn.net/babybin/article/details/111624905

版权

Python 专栏收录该内容

18 篇文章 1 订阅

订阅专栏

本文介绍了Python的PriorityQueue用法，强调了优先级队列中元素比较的重要性。错误示例展示了当队列中元素包含不可比较的字典时会引发TypeError。正确使用PriorityQueue的方式包括：以元组形式提供优先级和数据，或创建自定义类并重写`__lt__`方法以确保元素间可比较。示例代码展示了这两种正确用法。

摘要由CSDN通过智能技术生成

PriorityQueue是优先级队列。越小的优先级越高，会被先取出。

下面的代码运行正常。

# 示例1
tsq = queue.PriorityQueue()    
tsq.put_nowait((0, '123', ['abc', 'efg'], 0))
tsq.put_nowait((0, '456', ['abc'], 0))

下面的代码运行报错。

# 示例2
tsq = queue.PriorityQueue()
tsq.put_nowait((0, '123', {"name":'abc', "age":'efg'}, 0))
tsq.put_nowait((0, '123', {"name":'abc'}, 0))

原因：
PriorityQueue是优先级队列，优先级排序使用了堆排序，这一操作会对压入的元素进行比较。示例2中压入的元素是tuple，两个tuple相比较是将tuple中元素逐个比较。示例2中压入的两个tuple的前两个元素都相同，第三个元素{"name":'abc', "age":'efg'}和{"name":'abc'}都是字典，字典是不能比较大小的。
所以会报TypeError: '<' not supported between instances of 'dict' and 'dict'的错误。

PriorityQueue的正确使用方式，应该是如下两种，使用tuple的第一个元素作为优先级数字，或者自定义类中重定义__lt__方法，使得类实例能够相互比较。

# 示例3
tsq = queue.PriorityQueue()
tsq.put_nowait((0,{'task_name': 'aaa'}))  # tuple包含两个元素，第一个是优先级，第二个是数据
tsq.put_nowait((1, {'task_name', 'bbbb'}))

# 示例4
import queue


class Task(object):
    def __init__(self, priority, name):
        self.priority = priority
        self.name = name

    def __str__(self):
        return "Task(priority={p}, name={n})".format(p=self.priority, n=self.name)

    def __lt__(self, other):
    	""" 定义<比较操作符。 """
        return self.priority < other.priority


tsq = queue.PriorityQueue()
tsq.put_nowait(Task(3, "task1"))  # 自定义的类定义了__lt__, 可以比较大小
tsq.put_nowait(Task(1, "task2"))
print(tsq.get())  # return: Task(priory=1, name=task2)