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

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注