Python中的迭代器

**

迭代器

**
迭代器是访问集合元素的一种方式。迭代器对象从集合中第一个元素开始访问，直到所有的元素被访问完结束。迭代器只能往前不会后退。
迭代器的两个方法：
iter()方法：返回迭代器对象本身
next()方法：返回容器的下一个元素，在结尾时引发StopIteration异常。
对于可迭代对象，可以使用内建函数iter()来获取它的迭代器对象：

list=[1,2]
iter=iter(list)
print iter.next()
print iter.next()
print iter.next()

打印出：
1
2
Traceback (most recent call last):
File “C:/Users/Administrator/Desktop/iter3.py”, line 5, in
print iter.next()
StopIteration

自定义迭代器

下面定义了一个MyRange类，这个类型中实现了iter()方法，通过这个方法返回对象本身作为迭代器对象；同时，实现了next()方法用来获取容器中的下一个元素，当没有可访问元素后，就抛出StopIteration异常。

class MyRange(object):
    def __init__(self,n):
        self.idx=0
        self.n=n
    def __iter__(self):
        return self
    def next(self):
        if self.idx<self.n:
            val=self.idx
            self.idx+=1
            return val
        else:
            raise StopIteration()
if __name__=="__main__":
    myRange=MyRange(3)
    for i in myRange:
       print i

打印结果为：
0
1
2
因为myRange这个对象就是一个可迭代对象，同时它本身也是一个迭代器对象，所以就出现了问题，他没有办法支持多次迭代，例如：

class MyRange(object):
    def __init__(self,n):
        self.idx=0
        self.n=n
    def __iter__(self):
        return self
    def next(self):
        if self.idx<self.n:
            val=self.idx
            self.idx+=1
            return val
        else:
            raise StopIteration()
if __name__=="__main__":
    myRange=MyRange(3)
    print myRange is iter(myRange)
    print [i for i in myRange]
    print [i for i in myRange]

打印结果为：
True
[0, 1, 2]
[]
为了解决上面的问题，可以分别定义可迭代类型对象和迭代器类型对象；然后可迭代类型对象的iter()方法可以获得一个迭代器类型的对象。

class Zrange:
    def __init__(self,n):
        self.n=n
    def __iter__(self):
        return MyRange(self.n)


class MyRange(object):
    def __init__(self,n):
        self.idx=0
        self.n=n
    def __iter__(self):
        return self
    def next(self):
        if self.idx<self.n:
            val=self.idx
            self.idx+=1
            return val
        else:
            raise StopIteration()
if __name__=="__main__":
    zrange=Zrange(3)
    print zrange is iter(zrange)
    print [i for i in zrange]
    print [i for i in zrange]

打印结果为：
False
[0, 1, 2]
[0, 1, 2]