在Python中，迭代器是一個(gè)很重要的東西。良好的迭代器定義，能讓代碼更加清晰。

定義一個(gè)迭代器需要實(shí)現(xiàn)兩個(gè)方法__iter__和__next__。

__iter__返回迭代器對象本身。它是必需的，以允許容器和迭代器在for .. in ..語句中使用。
__next__返回下一個(gè)元素。當(dāng)沒有元素可返回時(shí)，應(yīng)拋出StopIteration。

class X2:
    def __init__(self, n: int):
        self.list = [x*x for x in range(n)]
        self.__count = -1

    def __iter__(self):
        return self

    def __next__(self):
        self.__count += 1
        try:
            return self.list[self.__count]
        except IndexError:
            raise StopIteration


if __name__ == "__main__":
    for each in X2(10):
        print(each)

但這樣定義似乎有些太麻煩了，Python提供了一個(gè)更加簡略的方法：

def X2(n: int):
    for i in range(n):
        yield i*i


if __name__ == "__main__":
    for each in X2(10):
        print(each)

事實(shí)上，如下代碼

    for each in X2(10):
        print(each)

與

    t = X2(10)
    while True:
        try:
            each = t.__next__()
        except StopIteration:
            break
        print(each)

是等價(jià)的。

能看出，for .. in ..的運(yùn)行機(jī)制，就是不斷地調(diào)用被迭代對象的.__next__()方法，直到捕獲到StopIteration異常。

明白了這一點(diǎn)，很容易就能知道，Python的死循環(huán)寫法至少有兩種：while True:和使用for .. in ..迭代一個(gè)擁有__iter__和__next__方法的對象，并且這個(gè)對象的__next__方法永遠(yuǎn)不會拋出StopIteration異常。

附：next(g)和g.__next__()都是調(diào)用了g的__next__()方法，只不過next(g)可能看起來更友好一點(diǎn)。