在前面讲迭代器和生成器原理的时候,就已经说过__iter__方法和__next__方法的作用了,在这里就不做重复描述了。

在这补一个实现迭代器协议的例子。

例1:(在没有抛出StopIteration这个异常之前,这个迭代器会无限的生成值)

class c1:

    def __init__(self,start):

        self.start = start

    def __iter__(self):

        return self

    def next(self):

        self.start += 1

        return self.start

o1 = c1(10)

for i in o1:

    print i


例2:(抛出stopiteration异常之后)

class c1:

    def __init__(self,start,stop):

        self.start = start

        self.stop = stop

    def __iter__(self):

        return self

    def next(self):

        if self.start >= self.stop:

            raise StopIteration

        n = self.start

        self.start += 1

        return n

o1 = c1(1,10)

for i in o1:

    print i

输出:

1

2

3

4

5

6

7

8

9


斐波那契数列:

class Fib:

    def __init__(self):

        self._a=0

        self._b=1

    def __iter__(self):

        return self

    def __next__(self):

        self._a,self._b=self._b,self._a + self._b

        return self._a

f1=Fib()