十一、生成器和迭代器 5.生成器应用

生成器应用

如果推算的算法比较复杂,用类似list推导式的for循环无法实现的时候,可以用函数来实现

  • 迭代器中return和send()的使用
# 斐波那契数列
# 实现:使用生成器返回斐波那契数列的指定位

# 创建生成器
def Fibonacci(n):
    a = 1
    b = 1
    current_index = 0

    # 循环生成数据
    print('----- no run -----')
    while current_index < n:
        data = a
        a , b = b , a+b
        current_index += 1
        print('----- yield 循环内部 -----')

        # 接收数据
        s = yield data
        print('----- yield 下一次循环开始 -----')

        # 根据条件,控制return终止生成器
        if s == 1:
            return '终止生成器'

if __name__ == '__main__':
    fib = Fibonacci(15)
    print(fib,type(fib))
    # res = list(fib)
    # print(res)
    print(next(fib))
    try:
        print(next(fib))
        print(next(fib))

        # 唤醒并传递参数到yield
        print(fib.send(1))
    except Exception as e:
        print(e)


<generator object Fibonacci at 0x0000002F49281148> <class 'generator'>
----- no run -----
----- yield 循环内部 -----
1
----- yield 下一次循环开始 -----
----- yield 循环内部 -----
1
----- yield 下一次循环开始 -----
----- yield 循环内部 -----
2
----- yield 下一次循环开始 -----
终止生成器
上一篇:P6835 [Cnoi2020]线形生物


下一篇:2014 hack.lu——oreo