一个自定义的迭代器

一个类如何成为迭代器类型，请看官方PEP说明：

即必须实现两个方法（或者叫两种协议）：__iter__ , __next__

下面编写一个迭代器类：

class YourRange():
    def __init__(self, start, end):
        self.value = start
        self.end = end

    # 成为迭代器类型的关键协议
    def __iter__(self):
        return self

    # 当前迭代器状态(位置)的下一个位置
    def __next__(self):
        if self.value >= self.end:
            raise StopIteration

        cur = self.value
        self.value += 1
        return cur

使用这个迭代器：

yr = YourRange(5, 12)
for e in yr:
    print(e)

迭代器实现__iter__ 协议，它就能在 for 上迭代，参考官网PEP解释：

文章最后提个问题，如果此时运行：

next(yr)

会输出 5， 还是报错？

如果 yr 是 list，for 遍历后，再 next(iter(yr)) 又会输出什么？

如果能分清这些问题，恭喜你，已经真正理解迭代器迭代和容器遍历的区别。如果你还拿不准，欢迎留言交流。

你点的每个“在看”，我都认真当成了喜欢