如何優雅地使用Python迭代器

Question

我想更多地使用循環迭代器，因爲我聽說它比索引循環更快。我不確定的一件事是如何很好地處理序列的結尾。我能想到的方式是使用try和except StopIteration，這對我來說看起來很醜。

更具體地說，假設我們被要求打印兩個排序列表a和b的合併排序列表。我會寫下面的內容

aNull = False 
I = iter(a) 
try: 
    tmp = I.next() 
except StopIteration: 
    aNull = True 

for x in b: 
    if aNull: 
     print x 
    else: 
     if x < tmp: 
      print x 
     else: 
      print tmp,x 
      try: 
       tmp = I.next() 
      except StopIteration: 
       aNull = True 

while not aNull: 
    print tmp 
    try: 
     tmp = I.next() 
    except StopIteration: 
     aNull = True 

你會怎麼編碼使它更整潔？

Answer 1

我認爲更對稱地處理a和b會使讀起來更容易。此外，使用在Python 2.6內置next函數的默認值避免了需要處理StopIteration：

def merge(a, b): 
    """Merges two iterators a and b, returning a single iterator that yields 
    the elements of a and b in non-decreasing order. a and b are assumed to each 
    yield their elements in non-decreasing order.""" 

    done = object() 
    aNext = next(a, done) 
    bNext = next(b, done) 

    while (aNext is not done) or (bNext is not done): 
     if (bNext is done) or ((aNext is not done) and (aNext < bNext)): 
      yield aNext 
      aNext = next(a, done) 
     else: 
      yield bNext 
      bNext = next(b, done) 

for i in merge(iter(a), iter(b)): 
    print i 

以下功能概括爲任意多的迭代器的工作方式。

def merge(*iterators): 
    """Merges a collection of iterators, returning a single iterator that yields 
    the elements of the original iterators in non-decreasing order. Each of 
    the original iterators is assumed to yield its elements in non-decreasing 
    order.""" 

    done = object() 
    n = [next(it, done) for it in iterators] 

    while any(v is not done for v in n): 
     v, i = min((v, i) for (i, v) in enumerate(n) if v is not done) 
     yield v 
     n[i] = next(iterators[i], done) 

Answer 2

你錯過了迭代器的全部觀點。您不需要手動撥打I.next()，只需通過I進行重複。

for tmp in I: 
    print tmp 

編輯

要合併兩個迭代器，該itertools模塊中使用非常方便的功能。你想要的是可能izip：

merged = [] 
for x, y in itertools.izip(a, b): 
    if x < y: 
     merged.append(x) 
     merged.append(y) 
    else: 
     merged.append(y) 
     merged.append(x) 

再次編輯

正如在評論中指出，這實際上不會工作，因爲有可能是從列表比下一個更小的多個項目列表b中的項目。但是，我意識到還有另一個內置函數可以處理這個問題：heapq.merge。

Answer 3

函數sorted適用於列表和迭代器。也許這不是你想要的，但下面的代碼有效。

 

a.expand(b) 
print sorted(iter(a))