python __iter__&__next__

iter(object)会调用object的__iter__方法,所以需要在__iter__方法中返回一个可迭代的对象。
可迭代对象应该实现__next__方法,供next方法来调用,以实现迭代效果。

下例中iter(A())调用了A中的__iter__方法,返回了B的对象实例。
next(obj)调用B对象实例中的__next__方法来进行迭代

class A(object):
    def __iter__(self):
        return B()

class B(object):
    def __init__(self):
        self.data = 1
    def __next__(self):
        self.data += 1
        return self.data



obj = iter(A())

next(obj)

应用1:实现range

class MyRange(object):
    def __init__(self, end):
        self.start = 0
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.start < self.end:
            ret = self.start
            self.start += 1
            return ret
        else:
            raise StopIteration

a = MyRange(4)

for i in a:
    print(i)

参考:
https://blog.csdn.net/xmlbm/article/details/99985678

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值