class Classmate():
def __init__(self):
self.name = list()
self.current = 0
#print(self.name)
def add(self,name):
self.name.append(name)
#print(self.name)
def __iter__(self):
return self#一定要返回为self,或者return 一个含有__iter__和__next__函数的对象
def __next__(self):
#当for循环调用时不会自动增加列表中的值,需要加一个属性进行增加
#
if self.current<len(self.name):
ret = self.name[self.current]
self.current += 1
return ret
else:
raise StopIteration
#当调用时将抛出异常,超过数组的长度
#方法解决:超过数据长度时,主动抛出异常使for循环停止
classmate = Classmate()
classmate.add("张三")
classmate.add("张三1")
classmate.add("张三2")
for i in classmate:
#for循环只能用于可迭代的对象,判断对象是否可迭代,1:需要有__iter__方法以及 __next__方法,其中__iter__方法一定要实现,__next__方法为for循环调用时使用的值
print(i)