4.1 手動(dòng)遍歷迭代器

問(wèn)題

你想遍歷一個(gè)可迭代對(duì)象中的所有元素，但是卻不想使用for循環(huán)。

解決方案

為了手動(dòng)的遍歷可迭代對(duì)象，使用 next() 函數(shù)并在代碼中捕獲 StopIteration 異常。比如，下面的例子手動(dòng)讀取一個(gè)文件中的所有行：

def manual_iter():
    with open('/etc/passwd') as f:
        try:
            while True:
                line = next(f)
                print(line, end='')
        except StopIteration:
            pass

通常來(lái)講，StopIteration 用來(lái)指示迭代的結(jié)尾。然而，如果你手動(dòng)使用上面演示的 next() 函數(shù)的話，你還可以通過(guò)返回一個(gè)指定值來(lái)標(biāo)記結(jié)尾，比如None。下面是示例：

with open('/etc/passwd') as f:
    while True:
        line = next(f)
        if line is None:
            break
        print(line, end='')

討論

大多數(shù)情況下，我們會(huì)使用for循環(huán)語(yǔ)句用來(lái)遍歷一個(gè)可迭代對(duì)象。但是，偶爾也需要對(duì)迭代做更加精確的控制，這時(shí)候了解底層迭代機(jī)制就顯得尤為重要了。

下面的交互示例向我們演示了迭代期間所發(fā)生的基本細(xì)節(jié)：

>>> items = [1, 2, 3]
>>> # Get the iterator
>>> it = iter(items) # Invokes items.__iter__()
>>> # Run the iterator
>>> next(it) # Invokes it.__next__()
1
>>> next(it)
2
>>> next(it)
3
>>> next(it)
Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
StopIteration
>>>

本章接下來(lái)幾小節(jié)會(huì)更深入的講解迭代相關(guān)技術(shù)，前提是你先要理解基本的迭代協(xié)議機(jī)制。所以確保你已經(jīng)把這章的內(nèi)容牢牢記在心中。