迭代器的实例对象

实例一：

#  写一个实现迭代器协议的类 Primes 让此类可以生成从b开始的n个素数

# class Primes:
#     def __init__(self, b, n):
#        ....
#     ....
# for x in Primes(10, 4):
#     print(x)  # 11 13 17 19

class Primes:




    def __init__(self,m,n):  #这里面的m 表示开始的数字 n为所取的个数
        self.m=m
        self.n=n
    def __repr__(self):
        return 'Primes(%r,%r)'%(self.m,self.n)


    def __iter__(self): #可迭代对象
        return PrimesNext(self.m,self.n) #返回迭代器




class PrimesNext:
    @staticmethod #静态函数  判断素数
    def primes(m):
        for i in range(2,m):
            if  m%i==0:
                return False
        return True
    def __init__(self,m1,n1):
       
        self.sur=0# 统计素数出现的次数
        self.m1=m1
        self.n1=n1
    def __next__(self):
        if self.sur>self.n1:#条件满足进行退出
            raise StopIteration
        #这里进行遍历m数字后面的数字然后进行判断
        while 1: #这里面进行无线循环
            
            if self.primes(self.m1):
                self.sur+=1  #计数加1
                v=self.m1 #把得到的值 赋值保存，继续进行下面的加运算
                self.m1+=1#为l在退出的时候能够进行下次的迭代，为进行下一次判断做准备
                return v
            self.m1+=1 #进行下一次的判断
for x in Primes(10,4):
    print(x)

实例二：# 写一个类,Fibonacci 实现迭代器协议,此类的对象可以作为可迭代对象生成相应的斐波那契数
#      1 1 2 3 5 8 ....


#     class Fibonacci:
#         def __init__(self, n):
#            ...


#     实现如下操作:
#     for x in Fibonacci(10):
#         print(x)
#     L = [ x for x in Fibonacci(30)]
#     print(sum(Fibonacci(25)))
#       (需要实现迭代器协议)

代码如下：

class Fibonacci:
    def __init__(self,data):
        self.data=data
    # def __rper__(self):
    #     return 'Fibonacci(%r)'%self.data
    def __iter__(self):#创建可迭代对象
        return FibonacciNext(self.data)#返回迭代器




class FibonacciNext:
    def __init__(self,next_data):
        self.next_data=next_data
        self.cur=0#进行可迭代对象元素长度的统计
        self.a=0 #设置数列开始值
        self.b=1#设置数列开始值   从0 开始 
     
    def __next__(self):
        if self.cur>self.next_data:
            raise StopIteration
        self.a,self.b=self.b,self.a+self.b #非波数列的机制
        self.cur+=1
        return self.b


for x in Fibonacci(10):
    print(x)