extracting a heapq in a for loop - there must be more elegant solution
Helmut Jarausch
jarausch at igpm.rwth-aachen.de
Tue Dec 3 07:18:59 EST 2013
Hi,
I'd like to extracted elements from a heapq in a for loop.
I feel my solution below is much too complicated.
How to do it more elegantly?
I know I could use a while loop but I don't like it.
Many thanks for some lessons in Python.
Here is my clumsy solution
from heapq import heappush, heappop
# heappop raises IndexError if heap is empty
H=[]
for N in 'H','C','W','I' :
heappush(H,N)
# how to avoid / simplify the following function
def in_sequence(H) :
try :
while True :
N= heappop(H)
yield N
except IndexError :
raise StopIteration
# and here the application:
for N in in_sequence(H) :
print(N)
More information about the Python-list
mailing list