Python自己实现迭代器

这节课的practice是自己写一个迭代器

要求：1.实现next功能 2.实现add功能

#将其实现写进一个类
class custom_itr(object):
    def __init__(self):#构造函数
        self.container=[]#使用一个list存放元素
        self.index=0#使用一个整型index表示下标

    def __add__(self, value):
        self.container.append(value)#使用list自带的append实现尾增加元素

    def __next__(self):
        if self.index<len(self.container):#判断下标指针是否合理
            elem=self.container[self.index]#给临时变量elem赋值
            self.index+=1#下标移动
            return elem
        else:
            raise Exception #移动失败异常处理

    def __iter__(self): #实现iter操作
        return self

    def distroyed(self):#实现销毁操作
        del self.container
        self.index=-1


newl=custom_itr()
newl.__add__("23333")
newl.__add__("23333")#插值

print(newl.__next__())#移动读取
print(newl.__next__())

运行结果：

23333
23333