Python 迭代器与生成器

迭代器 生成器

01 迭代器

python中有些对象是可迭代的，我们可以自己创建迭代器使自定义的类可迭代，从而可以使用for循环输出

1. for循环实现过程
   1. 判断xxx是否可迭代
   2. 调用iter函数得到xxx对象的iter方法的返回值
   3. iter方法的返回值是一个迭代器
   4. 迭代器中有iter,next就可以使用for循环
   5. 每次for循环返回next的返回值

2. 代码示例

class Classmate():
    def __init__(self):
        self.names=list()
    def add(self,name):
        self.names.append(name)

    # 如果想要对象可以迭代，必须使用__iter__方法
    def __iter__(self):
        return Classiterator(self)  # 将自身的引用传给迭代器

# 迭代器中有__iter__，__next__就可以使用for循环
class Classiterator():
    def __init__(self,class_obj):
        self.iterator_obj=class_obj  # 接受类的引用，从而可以使用类中的值
        self.current_num=0  # 记录取值的位置

    def __iter__(self):  # 使迭代器成为可迭代的
        pass

    def __next__(self):  # 每次for循环时返回next中的返回值
        if self.current_num

• 代码中创建了两个类，classmate和它对应的迭代器
• 要想使一个类可迭代，就要将其自身的引用传给迭代器
• 迭代器通过_init_接收传递的参数
• 迭代器中也需要有_iter_使得迭代器可迭代
• 使用for循环输出可迭代对象时，每次返回_next_中的返回值

3.斐波那契数列迭代器
将类与迭代器整合在了一起

class Fibonacci():
    def __init__(self,n):
        self.n=n
        self.current_num=0

        self.a=1
        self.b=1

    def __iter__(self):
        return self

    def __next__(self):
        if self.current_num < self.n:
            result = self.a
            self.a,self.b=self.b,self.a+self.b
            self.current_num+=1
            return result

        else:
            raise StopIteration

fibo=Fibonacci(10)

for num in fibo:
    print(num)

02 生成器

生成器是一种特殊的迭代器，使用起来更加简单

1. 简单生成列表与表达式的方法

list1=[x for x in range(10)]  # 生成一个列表
gen1=(x for x in range(10))  # 生成一个生成器

print(list1)
print(next(gen1))

2. 生成器函数
   只要函数中有yield，它就是一个生成器

   使用生成器输出斐波那契数列

def create_fibo(n):
    i=0
    a, b=1, 1

    while i

3. 启动生成器的方法

   1. 使用迭代的方法

   for num in nums:
    print(num)
   

   2. 使用next

   while True:
     try:
       result=next(nums)
       print(result)
       
     except Exception as result:
       print(result.value)
       break
   

   3. 使用send

   ret=nums.send(None)
   print(ret)
   print('==============')
   ret=nums.send('hahaha')
   print(ret)
   print(ret)
   ret=nums.send('hahaha')
   
   '''
   1
   ==============
   hahaha
   1
   1
   hahaha
   '''
   

   • send可以传递参数，默认为None
   • 参数是yield的返回值，如调用ret = yield a时就返回了send传递的参数值
   • 第一次调用send时，参数必须为None，或者使用next启动