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