如何在Python中使用heapq模块实现一个带优先级的任务队列？-Python教程-PHP中文网

python-parallelization

将顺序 Python 代码转换为并行/并发实现，用于在需要并行化 Python 代码、提升性能时使用。

下载

直接用heapq实现优先级队列可行但需手动维护可比较性；因heapq.heappush()依赖元素比较，而字典或未实现__lt__的自定义类实例不可比较，会触发typeerror。

如何在python中使用heapq模块实现一个带优先级的任务队列？

直接用 heapq 实现带优先级的任务队列是可行的，但必须手动维护任务的可比较性——Python 的 heapq 不支持直接 push 自定义对象，除非它们能被正确比较。

为什么不能直接 push 字典或普通类实例？

因为 heapq.heappush() 依赖元素间的 比较，而字典、未定义 <code>__lt__ 的类、或包含不可比较字段（如 None、函数）的对象会抛出 TypeError: '。

错误示例：heapq.heappush(q, {'priority': 2, 'task': 'send_email'}) → 直接报错
即使所有字典都有 priority，Python 也不按该字段隐式排序
元组可以，但需确保第一个元素可比较且能表达优先级逻辑（比如负数模拟最大堆）

推荐写法：用元组封装优先级 + 任务数据

最轻量、最可靠的方式是把任务包装成 (priority, count, task) 元组，其中 count 是单调递增计数器，用于打破优先级相同时的比较歧义（避免比较到不可比较的 task）。

import heapq
<p>class PriorityQueue:
def <strong>init</strong>(self):
self._queue = []
self._count = 0  # 保证相同 priority 下顺序稳定</p><pre class="brush:php;toolbar:false;"><code>def push(self, item, priority):
    # 用负 priority 实现最大堆？不，这里用最小堆：priority 越小越先执行
    heapq.heappush(self._queue, (priority, self._count, item))
    self._count += 1

def pop(self):
    if self._queue:
        return heapq.heappop(self._queue)[-1]  # 只返回 task
    raise IndexError("pop from empty priority queue")</code>

priority 是数值型（int 或 float），越小优先级越高
self._count 防止两个任务 priority 相同时，Python 去比较 item（可能失败）
不要用 (-priority, ...) 模拟最大堆——容易混淆语义，且浮点精度可能引发意外排序

如果必须用自定义类，怎么让 heapq 接受它？

给类实现 __lt__（less-than），且只基于可比较字段（如 priority）定义逻辑，其他字段不参与比较。

class Task:
    def __init__(self, name, priority):
        self.name = name
        self.priority = priority
<pre class="brush:php;toolbar:false;"><code>def __lt__(self, other):
    return self.priority <p>q = []
heapq.heappush(q, Task("backup", 3))
heapq.heappush(q, Task("alert", 1))
print(heapq.heappop(q).name)  # 输出 "alert"
</p></code>