一个自定义的迭代器
一个类如何成为迭代器类型，请看官方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)) 又会输出什么？
如果能分清这些问题，恭喜你，已经真正理解迭代器迭代和容器遍历的区别。如果你还拿不准，欢迎留言交流。
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2020-05-15，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读