生成器应用
如果推算的算法比较复杂,用类似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 下一次循环开始 -----
终止生成器