yield关键字

2023-07-30 17:43:52 浏览数 (2)

在一个函数内实用yield关键字将函数变成生成器。

一个简单的例子:
代码语言:javascript复制
#  一个迭代器函数
def fun(n):
    for i in range(n):
        yield i
        print('--分割线--')

#  获取迭代器
a = fun(5)
#  使用next让迭代器继续执行
print(next(a))
print(next(a))
print(a.__next__())
print(a.__next__())

解释:

  1. 定义一个生成器a,运行完第三行停止
  2. 第一个next函数执行,接收第一个i,此时i=0,输出0,第三行后的代码不执行
  3. 第二个next函数执行,从第四行执行代码,输出分割线后,进入下一个for循环,接收第二个i,此时i=1,输出1,第三行后的代码不执行
  4. 第三个next方法执行,从第四行执行代码,输出分割线后,进入下一个for循环,接收第二个i,此时i=2,输出2,第三行后的代码不执行

抽象解释: 将yield看成两个部分,一:return i,二:一个锁,每次执行next函数后,程序运行在一和二之间。下一次调用next函数先开第二部分的锁,输出分割线后进入一个新的循环,再return一个新的i,被新的锁锁住。(第一次相当于没有锁,直接return后再被锁)

tip:next函数和__next__()方法效果相同

关闭生成器:
代码语言:javascript复制
#  关闭后无法用next进行迭代(用的少)
a.close()
向生成器发送数据:

改一下代码:

代码语言:javascript复制
#  一个迭代器函数
def fun(n):
    for i in range(n):
        k = yield i
        if k:
            print(k)
        print('--分割线--')

#  获取迭代器
a = fun(5)
print(next(a))
#  发送一个字符
print(a.send('a'))
print(a.__next__())
print(a.__next__())

可以看到a被输出了,不影响其他代码运行(send后同样进行一次迭代,结果和next相同)

传入一个异常:
代码语言:javascript复制
#  此时生成器会抛出一个错误(可以用try语句接收)
a.throw(Exception)

#  一般这样接收
try:
    yield 1
except Exception:
    yield 'Exception'
finally:
    print('finally')

0 人点赞