欢迎来到Python教程自学网!
当前位置: 首页 > Python基础教程 > python迭代器怎样自定义类

python迭代器怎样自定义类

   

python迭代器怎样自定义类

1、说明

迭代器还具有迭代用户定制类别的能力。迭代对象需要支持两种方式:_iter__()和next(),前者返回迭代本身,后者返回下一个元素。

2、实例

class example(object):
    def __init__(self,num):
        self.num=num
    def __iter__(self):
        return self
    def __next__(self):
        if self.num <= 0:
            raise StopIteration
        tmp = self.num
        self.num -= 1
        return tmp
    
a = example(3)
print(a.__next__())
print(a.__next__())
print(a.__next__())
print(a.__next__())