迭代器只能迭代一次嗎？

Question

請看下面的例子：

def foo(iterator): 
    return sum(iterator)/max(iterator) 

是否安全，重複使用相同的迭代器兩次？

Answer 1

不，它不安全。迭代器不是序列。下面是使用一臺發電機與foo()功能的情況發生，這本身就是一個迭代器：

>>> foo(x for x in range(10)) 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "<stdin>", line 1, in foo 
ValueError: max() arg is an empty sequence 

這是因爲迭代器已經在其結束時後sum()完成自己的工作，所以max()正從它沒有其他項目。通常，重置迭代器是不可能的，以便它可以再次循環。爲了使foo()功能很好地支持迭代器，它必須通過保存的物品從iterator重寫，這樣迭代通過循環只有一次，例如到一個臨時tuple或list：

def foo(iterator): 
    iterable = list(iterator) 
    return sum(iterable)/max(iterable) 

，或者，如果iterator產生了大量的項目，由仔細使用for循環處理它：

def foo(iterator): 
    # allows iterables like lists or tuples to be passed as arguments 
    iterator = iter(iterator) 

    try: 
     max_ = next(iterator) 
     sum_ = max_ 
    except StopIteration: 
     # iterator yields no items, return some default value 
     return 0.0 

    for element in iterator: 
     sum_ += element 
     max_ = max(max_, element) 

    return sum_/max_ 

這將產生正確的結果：

>>> foo(x for x in range(10)) 
5.0