yield关键字的说明
yield
是 Python 中的一个关键字,它通常与生成器函数一起使用。yield
就是保存当前程序执行状态。你用 for 循环的时候,每次取一个元素的时候就会计算一次。用 yield
的函数 叫 generator
,和 iterator
一样,它的好处是不用一次计算所有元素,而是用一次算一次,可以节省很多空间。generator
每次计算需要上一次计算结果,所以用 yield
,否则一 return
,上次计算结果就没了。
yield
可以简单理解为return
操作,但和return
又有很大的区别,执行完return
,当前函数就终止了,函数内部的所有数据,所占的内存空间,全部都没有了。而yield
在返回数据的同时,还保存了当前的执行内容,当你再一次调用这个函数时,他会找到你在此函数中的yield
关键字,然后从yield
的下一句开始执行。
当有多个返回值时,用 return
全部一起返回了,需要单个逐一返回时可以用 yield
。
带有yield
的函数在Python中被称之为generator
(生成器),也就是说,当你调用这个函数的时候,函数内部的代码并不立即执行 ,这个函数只是返回一个生成器(Generator Iterator)。
示例代码1:
def generator():
for i in range(10):
yield i * i
gen = generator()
print(gen)
我们可以看出,当你调用这个函数的时候,函数内部的代码并不立即执行 ,这个函数只是返回一个生成器(Generator Iterator)。
示例代码2:
def generator():
for i in range(10):
yield i * i
gen = generator()
print(gen)
print("first:")
print(next(gen))
print("second:")
print(next(gen))
注意:
- 在函数第一次调用
next(gen)
函数时,generator函数从开始执行到yield
,并返回yield
之后的值。 - 在函数第二次调用
next(gen)
函数时,generator函数从上一次yield
结束的地方继续运行,直至下一次执行到yield
的地方,并返回yield
之后的值。依次类推。
yield代码说明:
- 代码执行到
yield
会暂停,然后把结果返回出去,下次启动生成器会在暂停的位置继续往下执行 - 生成器如果把数据生成完成,再次获取生成器中的下一个数据会抛出一个StopIteration 异常,表示停止迭代异常
- while 循环内部没有处理异常操作,需要手动添加处理异常操作
- for 循环内部自动处理了停止迭代异常,使用起来更加方便,推荐大家使用。
实际yield
关键词的使用
def my_generator():
yield 1
yield 2
yield 3
my_generator
是一个生成器函数,它使用 yield
关键字返回了三个值。当 my_generator
函数被调用时,它不会立即执行,而是返回一个生成器对象。每次调用生成器对象的 __next__()
方法时,函数会从上一次暂停的位置继续执行,直到遇到下一个 yield
关键字。
可以使用 for
循环来迭代生成器对象,从而逐个获取生成器函数返回的值:
for value in my_generator():
print(value)
这将依次输出 1
、2
和 3
。
函数中有一个或者多个yield关键字
示例代码1:函数中有一个yield关键字
def mygenerater(n):
for i in range(n):
print('开始生成...')
yield i
print('完成一次...')
if __name__ == '__main__':
g = mygenerater(2)
# 获取生成器中下一个值
result = next(g)
print(result)
result = next(g)
print(result)
示例代码2:函数中有多个yield关键字
def mygenerater(n):
for i in range(n):
print('开始生成...')
yield i
print('完成一次...')
yield i+100
print('完成二次...')
print("生成一轮结束")
if __name__ == '__main__':
g = mygenerater(3)
# 获取生成器中下一个值
result = next(g)
print(result)
result = next(g)
print(result)
不论是有一个或者多个yield关键字,一次只能返回一个,多个yield关键字返回的内容轮流一次返回。
yield from
语法
除了 yield
关键字,Python 还提供了 yield from
语法,它可以简化生成器函数的定义。yield from
可以将一个嵌套的生成器的值逐个返回给调用者,而不需要手动遍历嵌套生成器的每个值。例如:
def my_generator():
yield from [1, 2, 3]
yield from (4, 5, 6)
yield from my_other_generator()
def my_other_generator():
yield "a"
yield "b"
yield "c"
在这个例子中,my_generator
函数使用 yield from
将一个列表、一个元组和另一个生成器的值逐个返回给调用者。调用 my_generator
函数时,它会返回一个生成器对象,该对象将逐个返回下列值:1
、2
、3
、4
、5
、6
、"a"
、"b"
和 "c"
。