即必须实现两个方法(或者叫两种协议):__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)) 又会输出什么?
如果能分清这些问题,恭喜你,已经真正理解迭代器迭代和容器遍历的区别。如果你还拿不准,欢迎留言交流。
本文分享自 程序员郭震zhenguo 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!