python迭代器的用法

python中可迭代对象是指实现了方法__iter__的对象，方法__iter__返回一个对象，可以是self，也可以是其他对象。

若方法__iter__返回的对象包含方法__next__，则该对象就是一个迭代器。方法__next__指每次对同一个对象调用该方法时，标记并且返回下一个值。

因为方法__next__总是指向下一个值，所以如果值用完了，就会返回None，此时需要手动引发StopIteration异常，否则会无限返回None，进入死循环。

综上所述，对于一个对象，如果同时包含方法__iter__和方法__next__，且__iter__返回self，那么该对象就是一个只能迭代一轮的迭代器，原因在于for循环的机制。

for循环的工作机理：

1. 在 for 语句内部先调用了__iter__ 方法，返回结果是一个迭代器，__iter__方法只会调用一次。

2. 然后对迭代器调用__next__方法，并将其返回值赋给循环变量。

3. 之后，会执行 for 循环中关联的语句块。

4. 在 __next__ 方法抛出 StopIteration 之前会一直重复执行第 2，3 步。

5. 一旦 __next__抛出 StopIteration，控制器会跳转到 else 子句（如果存在）并执行与 else 关联的语句块。

注意：如果在步骤 3 中，for 循环语句遇到了 break 语句，则跳过 else 代码块。

下面是一个返回10个斐波那契数的迭代器，该迭代器对象只能迭代一次

class fbs:
    def __init__(self):
        self.a=0
        self.b=1
        self.i=0
    def __next__(self):
        if(self.i<10):
            self.i = self.i + 1
            self.a,self.b=self.b,self.a+self.b
            return self.a
        else:
            raise StopIteration
    def __iter__(self):
        return self

>>> a=fbs()
>>> list(a)
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
>>> list(a)
[]

为了获得可以在for循环中重复迭代的迭代器，可以在__iter__方法中返回一个新的对象，而不再是自身。代码如下，仅仅将return self改为return fbs()

class fbs:
    def __init__(self):
        self.a=0
        self.b=1
        self.i=0
    def __next__(self):
        if(self.i<10):
            self.i = self.i + 1
            self.a,self.b=self.b,self.a+self.b
            return self.a
        else:
            raise StopIteration
    def __iter__(self):
        return fbs()

>>> a=fbs()
>>> list(a)
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
>>> list(a)
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]

也可以在另一个对象中实现__iter__方法

class fbs:
    def __init__(self):
        self.a=0
        self.b=1
        self.i=0
    def __next__(self):
        if(self.i<10):
            self.i = self.i + 1
            self.a,self.b=self.b,self.a+self.b
            return self.a
        else:
            raise StopIteration
class bbb:
    def __iter__(self):
        return fbs()

>>> a=bbb()
>>> list(a)
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]
>>> list(a)
[1, 1, 2, 3, 5, 8, 13, 21, 34, 55]

参考文章：http://t.csdn.cn/3uagR