python 迭代器

Posted 2020-08-09

迭代器：是一个实现了迭代器协议的对象，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]