【generator101】 – yield from
admin
2023-07-31 01:50:19
0

在python 3.3里,generator新增了一个语法 yield from
这个yield from的作用是什么?看下面两段对比的代码:

def zero_to_nine():
    for i in range(10):
        yield i

def wrap_generator():
    for i in zero_to_nine():
        yield i

print(list(zero_to_nine()))
print(list(wrap_generator()))

可以看到一个问题,如果一个函数内部调用了一个generator,它自身如果还希望是以generator的形式返回值的话,就需要用for循环重新yield一次。
使用yield from可以简化这个委托给子generator的过程

def zero_to_nine():
    for i in range(10):
        yield i

def wrap_generator():
    yield from zero_to_nine()

print(list(zero_to_nine()))
print(list(wrap_generator()))

如果只是这么一点写法上的区别的话,也就不劳烦BDFL添加新的语法了。我们可以看一下wrap_generator的bytecode

              0 LOAD_GLOBAL              0 (zero_to_nine) 
              3 CALL_FUNCTION            0 (0 positional, 0 keyword pair) 
              6 GET_ITER             
              7 LOAD_CONST               0 (None) 
             10 YIELD_FROM           
             11 POP_TOP              
             12 LOAD_CONST               0 (None) 
             15 RETURN_VALUE         

可见YIELD FROM是由cPython内部支持的,其实现原理上就避免了栈进栈出的消耗,直接由最内层的frame返回(yield)值。
另外YIELD FROM可以实现外部直接向最内层的generator传递值,比如

def i_yield_whatever_input_is():
    input = None
    while True:
        input = yield input

def wrap_generator():
    yield from i_yield_whatever_input_is()

gen = wrap_generator()
print(gen.send(None))
print(gen.send(1))
print(gen.send(2))
print(gen.send(3))

这段代码的输出是

None
1
2
3

这样send传值的方式,在用for循环重新yield的模式下是无法实现的。这也就是tulip必须使用yield from,而不能使用yield的原因。

相关内容

热门资讯

Mobi、epub格式电子书如... 在wps里全局设置里有一个文件关联,打开,勾选电子书文件选项就可以了。
定时清理删除C:\Progra... C:\Program Files (x86)下面很多scoped_dir开头的文件夹 写个批处理 定...
scoped_dir32_70... 一台虚拟机C盘总是莫名奇妙的空间用完,导致很多软件没法再运行。经过仔细检查发现是C:\Program...
500 行 Python 代码... 语法分析器描述了一个句子的语法结构,用来帮助其他的应用进行推理。自然语言引入了很多意外的歧义,以我们...
小程序支付时提示:appid和... [Q]小程序支付时提示:appid和mch_id不匹配 [A]小程序和微信支付没有进行关联,访问“小...
pycparser 是一个用... `pycparser` 是一个用 Python 编写的 C 语言解析器。它可以用来解析 C 代码并构...
微信小程序使用slider实现... 众所周知哈,微信小程序里面的音频播放是没有进度条的,但最近有个项目呢,客户要求音频要有进度条控制,所...
65536是2的几次方 计算2... 65536是2的16次方:65536=2⁶ 65536是256的2次方:65536=256 6553...
Apache Doris 2.... 亲爱的社区小伙伴们,我们很高兴地向大家宣布,Apache Doris 2.0.0 版本已于...
项目管理和工程管理的区别 项目管理 项目管理,顾名思义就是专注于开发和完成项目的管理,以实现目标并满足成功标准和项目要求。 工...