python - 有没有办法检查 Python 中的优先级队列中是否存在优先级?
问题描述
我对 Python 比较陌生。我正在尝试使用queue.PriorityQueue()
class
并有一个问题。有没有办法检查是否存在特定的优先级?
例如,我有以下内容(priority, element)
:
(0,A), (1,B), (2,C), (2,D)
有没有办法检查优先级是否2
存在?(是的,因为C
和D
都是 2。)我可以获取特定优先级的元素吗?所以,如果我想要优先的元素,2
它会给我C
和D
?
我在优先队列上看到的唯一官方文档是:优先队列官方文档
这个类还有其他文档吗?喜欢我可以使用的方法吗?可用的类/字段的结构?
解决方案
我的理解是,您的最终目标是将具有相同优先级的任务放在列表中并一起返回。所以:
如果您查看queue.PriorityQueue
(我做过)的代码,您会发现它基于模块的heappush
andheappop
方法,该heapq
模块实现了堆队列算法。见heapq。如果您在此页面上进一步查看,它们甚至显示了如何使用 heapq 来实现优先级队列。这个实现比您需要的要复杂一些,因为它支持更改已添加任务的优先级的能力,并且它不能完全处理具有相同优先级的多个任务。但是这些改变很容易做到:
from heapq import heappush, heappop
class PriorityQueue:
def __init__(self):
self._pq = [] # list of entries arranged in a heap
self._priority_finder = {} # mapping of priority to entries
def add_task(self, task, priority=0):
'Add a new task'
# any tasks with this priority?
entry = self._priority_finder.get(priority)
if entry:
entry[1].append(task)
else:
entry = [priority, [task]]
self._priority_finder[priority] = entry
heappush(self._pq, entry)
def pop_task(self):
'Remove and return the lowest priority tasks. Raise KeyError if empty.'
if not self._pq:
raise KeyError('pop from an empty priority queue')
priority, tasks = heappop(self._pq)
del self._priority_finder[priority]
return priority, tasks
def __bool__(self):
'return True if any tasks on the queue'
return True if self._pq else False
pq = PriorityQueue()
pq.add_task('a', 4) # task 'a' with priority 4
pq.add_task('b', 2)
pq.add_task('c', 4)
pq.add_task('d', 2)
pq.add_task('e', 1)
while pq:
print(pq.pop_task())
印刷:
(1, ['e'])
(2, ['b', 'd'])
(4, ['a', 'c'])
推荐阅读
- .net - Azure 服务总线队列接收器是否有超时限制?
- python - Django / Pandas - 创建 Excel 文件并作为下载
- shell - grep 中的 --exclude-dir 选项无法按预期工作
- vb.net - 防止从 FolderBrowseDialog VB.Net 访问网络驱动器
- javascript - 数据存储在数组中,但对于特定元素,它在反应 js 中显示为未定义
- javascript - 使用 IF 评估特定字符串
- reactjs - 由于某种原因我无法运行这个反应应用程序,我不知道该怎么做
- java - JPA - 实体的属性类型取决于列值
- php - 在 Wordpress 中如何创建子菜单?
- python - 为什么 python-docx 只读取 docx 的第一段?