1 生成器
根据程序员制定的规则循环生成数据,当条件不成立时则生成数据结束。数据不是一次性全部生成处理,而是使用一个,再生成一个,可以节约大量的内存。
两种方式创建生成器
- 生成器推导式
- yield 关键字
2 生成器推导式方式创建生成器:
# 创建生成器
my_generator = (i * 2 for i in range(5))
print(my_generator)
# next获取生成器下一个值
# value = next(my_generator)
# print(value)
# 遍历生成器
for value in my_generator:
print(value)
执行结果:
<generator object <genexpr> at 0x000001AF8E9C9510>
0
2
4
6
8
说明:
- 获取生成器下一个值得方法是调用next()
- 实际上是一个迭代器,因此可用for循环获取生成器值
3 yield 关键字创建生成器:
- 只要在def函数里面看到有 yield 关键字那么就是生成器
def my_g():
for i in range(3):
yield i
for i in my_g():
print(i)
执行结果:
0
1
2
Process finished with exit code 0
说明:
- 获取下一个值时,会执行for循环到yield这一行,返回yield后面得值,并阻塞在这一行,获取下一个值时继续往下执行
- 生成器如果把数据生成完成,再次获取生成器中的下一个数据会抛出一个StopIteration 异常,表示停止迭代异常
- for 循环内部自动处理了停止迭代异常,相对于while语句,使用起来更加方便
def fibonacci(num):
a = 0
b = 1
# 记录生成fibonacci数字的下标
current_index = 0
while current_index < num:
result = a
a, b = b, a + b
current_index += 1
# 代码执行到yield会暂停,然后把结果返回出去,下次启动生成器会在暂停的位置继续往下执行
yield result
fib = fibonacci(5)
# 遍历生成的数据
for value in fib:
print(value)
执行结果:
0
1
1
2
3
Process finished with exit code 0
5 生成器的send方法
generator.send(value)
恢复执行并向生成器函数“发送”一个值。 value 参数将成为当前 yield 表达式的结果。 send() 方法会返回生成器所产生的下一个值,或者如果生成器没有产生下一个值就退出则会引发 StopIteration。 当调用 send() 来启动生成器时,它必须以 None 作为调用参数,因为这时没有可以接收值的 yield 表达式。
def test():
for i in range(5):
sig = yield i
print('send by:', sig)
a = test()
print(a.__next__()) # 运行生成器,在yield处停止
print(a.send('a')) # 将“a”传递给sig变量,并在下一个yield处停止,返回1
print(a.send('b')) # 将“b”传递给sig变量,并在下一个yield处停止,返回2
print(a.__next__()) #
执行结果:
0
send by: a
1
send by: b
2
send by: None
3