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__())

以上就是python迭代器自定义类的方法，希望对大家有所帮助。更多Python高级指路：python高级

本文教程操作环境：windows7系统、Python 3.9.1，DELL G3电脑。