手写一个python迭代器过程详解
来源:
中文源码网 浏览:113 次 日期:2024-04-27 08:49:42
【下载文档: 手写一个python迭代器过程详解.txt 】
手写一个python迭代器过程详解
分析
我们都知道一个可迭代对象可以通过iter()可以返回一个迭代器。
如果想要一个对象称为可迭代对象,即可以使用for,那么必须实现__iter __()方法。
在一个类的实例对象想要变成迭代器,就必须实现__iter__()和__next__()方法。
调用iter()时,在对象内部默认调用__iter__(),即__iter__()的返回值应该是一个迭代器。
for的每次循环中或者next()时,都是自动调用迭代器的__next__()方法,并有一个返回值。
实现
class Classmate:
def __init__(self):
self.names = []
self.num = 0
def add(self, name):
self.names.append(name)
def __iter__(self):
return self
def __next__(self):
if self.num < len(self.names):
ret = self.names[self.num]
self.num += 1
return ret
else:
raise StopIteration
c = Classmate()
c.add("张三")
c.add("李四")
c.add("王五")
for i in c:
print(i)
"""
张三
李四
王五
"""
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持中文源码网。