Python | yield关键字详解

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"。