python 迭代器

迭代器:是一个实现了迭代器协议的对象,Python中的迭代器协议就是有next方法的对象会前进到下一结果,而在一系列结果的末尾是,则会引发StopIteration。任何这类的对象在Python中都可以用for循环或其他遍历工具迭代,迭代工具内部会在每次迭代时调用next方法,并且捕捉StopIteration异常来确定何时离开。

迭代器对象要求支持迭代器协议的对象,在Python中,支持迭代器协议就是实现对象的__iter__()和next()方法。其中__iter__()方法返回迭代器对象本身;next()方法返回容器的下一个元素,在结尾时引发StopIteration异常。当我们使用for语句的时候,for语句就会自动的通过__iter__()方法来获得迭代器对象,并且通过next()方法来获取下一个元素。在Python中,for循环可以用于Python中的任何类型,包括列表、元祖等等,实际上,for循环可用于任何“可迭代对象”,这其实就是迭代器。

class Myrange():
    def __init__(self,n):
        self.id = n
        self.index = 0
    def __iter__(self):
        return self
    def next(self):
        if self.index < self.id:
            self.index += 1
            return self.index
        else:
            raise StopIteration

a = Myrange(10)
print list(a)


#output
#/usr/bin/python /Users/li/PycharmProjects/Nowcoder_Practice/tmp.py
#[1,2, 3,4, 5,6, 7,8, 9,10]
原文地址:https://www.cnblogs.com/lirunzhou/p/5881193.html