python之__next__和__iter__

十 __next__和__iter__实现迭代器协议

class Foo:

    def __init__(self,n):
        self.n = n

    def __iter__(self):
        return self

    def __next__(self):
        if self.n == 13:
            raise StopIteration
        self.n += 1
        return self.n

f1 = Foo(10)
# print(f1.__next__())
# print(f1.__next__())
# print(f1.__next__())
# print(f1.__next__())

# 迭代器协议之一：必须要有__iter__方法，之二：必须要有__next__方法
# 当遇到抛出StopIteration异常时，结束循环
for i in f1:  #f1.__iter__()  == iter(f1)
     print(i)

迭代器协议实现斐波那契数列

class Foo:

    def __init__(self,a,b):
        self.a = a
        self.b = b

    def __iter__(self):
        return self

    def __next__(self):
        if self.a > 100:
            raise StopIteration
        self.a,self.b = self.b,self.a+self.b
        return self.a

f1 = Foo(1,1)
print(f1.__next__())
print(f1.__next__())
print('-------------------------')
for i in f1:  # 会接着上一次开始的地方继续执行
    print(i)

'''
1
2
-------------------------
3
5
8
13
21
34
55
89
144
'''