Python : 迭代器Iterators

Python 迭代器对象需要支持两个方法.

iter(),返回迭代器对象自身. 用于for x in 中.
next(),放回迭代器的下一个值.如果没有下一个值可以返回,那么需要抛出StopIteration异常.

class Counter(object):
    def __init__(self, low, high):
        self.current = low
        self.high = high
    def __iter__(self):
        return self
    def __next__(self):
        if self.current > self.high:
            raise StopIteration
        else:
            self.current += 1
            return self.current - 1
>>> c = Counter(5, 8)
>>> for i in c:
        print(i, end=' ')
5 6 7 8
迭代器只能被使用一次.抛出StopIteration 异常后会持续抛出相同异常.
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容