timevs_priority_queue

timevs_priority_queue.py

import timeit
code_heapq = """
import heapq, random
l = [random.randint(1, 1000) for _ in range(1000)]
pque = []
for x in l: heapq.heappush(pque, x)
for _ in range(1000): heapq.heappop(pque)
"""
code_PriorityQueue = """
import queue, random
l = [random.randint(1, 1000) for _ in range(1000)]
pque = queue.PriorityQueue()
for x in l: pque.put(x)
for _ in range(1000): pque.get()
"""

print('{:20}{:.3f}s'.format('heapq:', timeit.timeit(stmt=code_heapq, number=1000)))
print('{:20}{:.3f}s'.format('queue.PriorityQueue:', timeit.timeit(stmt=code_PriorityQueue, number=1000)))