yield全面总结

最新推荐文章于 2023-01-26 18:04:33 发布
最新推荐文章于 2023-01-26 18:04:33 发布
阅读量138 收藏
点赞数
文章标签： python
原文链接：https://juejin.im/post/5aa7341ff265da23923600e0
版权

本文首发于知乎
yield生成器在python中使用广泛，更是python中协程的实现原理，有必要深入掌握。

本文分为如下几个部分

  • 简单yield的使用
  • yield空
  • yield from
  • send与yield赋值
  • return yield

简单yield的使用

下面是一个最简单的yield使用的例子

def myfun(total):
for i in range(total):
yield i
复制代码

定义了myfun函数，调用时比如myfun(4)得到的是一个生成器，生成器有3种调用方式

1.用next调用

>>> m = myfun(4)
>>> next(m)
0
>>> next(m)
1
>>> next(m)
2
>>> next(m)
3
>>> next(m)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
复制代码

每次调用一次next，函数都会执行到yield处停止，等待下一次next，next次数超出限制就会抛出异常

2.循环调用

for i in myfun(4):
print(i)
复制代码

运行结果为

0
1
2
3
复制代码

生成器是可迭代对象，可以用循环调用。循环调用就是最大限度地调用next，并返回每次next运行结果

这就是yield的最基本使用，下面来应用一下

python内置模块itertools中实现了一些小的生成器，我们这里拿count举例子，要实现下面这样的效果

>>> from itertools import count
>>> c = count(start = 5, step = 2)
>>> next(c)
5
>>> next(c)
7
>>> next(c)
9
>>> next(c)
11
......
复制代码

实现如下

def count(start, step = 1):
    n = 0
while True:
yield start + n
        n += step
复制代码

3.循环中调用next

下面循环中调用next，用StopIteration捕获异常并退出循环

>>> m = myfun(4)
>>> while True:
...     try:
...         print(next(m))
...     except StopIteration:
...         break
...
0
1
2
3
复制代码

yield空

yield空相当于一个中断器，循环运行到这里会中断，用于辅助其他程序的执行。也可以理解成返回值是None，我们来看下面这个例子

>>> def myfun(total):
...     for i in range(total):
...         print(i + 1)
...         yield
...
>>> a = myfun(3)
>>> next(a)
1
>>> next(a)
2
>>> next(a)
3
>>> next(a)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
>>> a = myfun(3)
>>> for i in a:
...     print(i)
...
1
None
2
None
3
None
复制代码

yield from

通过下面一个例子来展示yield from的用法

def myfun(total):
for i in range(total):
yield i
yield from ['a', 'b', 'c']
复制代码

用next调用结果如下

>>> m = myfun(3)
>>> next(m)
0
>>> next(m)
1
>>> next(m)
2
>>> next(m)
'a'
>>> next(m)
'b'
>>> next(m)
'c'
>>> next(m)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
复制代码

上面的函数相当于

def myfun(total):
for i in range(total):
yield i
for i in ['a', 'b', 'c']:
yield i
复制代码

下面我们也做一个小练习，实现itertools模块中的cycle，效果如下

>>> from itertools import cycle
>>> a = cycle('abc')
>>> next(a)
'a'
>>> next(a)
'b'
>>> next(a)
'c'
>>> next(a)
'a'
>>> next(a)
'b'
>>> next(a)
'c'
>>> next(a)
'a'
复制代码

实现如下

def cycle(p):
yield from p
yield from cycle(p)
复制代码

send与yield赋值

先讲send，首先明确一点，next相当于send(None)，还是看下面最简单的例子

>>> def myfun(total):
...     for i in range(total):
...         yield i
...
>>> a = myfun(3)
>>> a.send(None)
0
>>> a.send(None)
1
>>> a.send(None)
2
>>> a.send(None)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
复制代码

如果send()参数不是None呢？send()表示向这个生成器中传入东西，有传入就得有变量接着，于是引出了yield赋值，来看下面一个例子

def myfun(total):
for i in range(total):
        r = yield i
        print(r)
复制代码

运行如下

>>> a = myfun(3)
>>> a.send(1)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
TypeError: can't send non-None value to a just-started generator
>>> a.send(None)
0
>>> a.send(1)
1
1
>>> a.send(1)
1
2
>>> a.send(1)
1
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration
复制代码

上面运行结果显示

1.第一次send参数必须是None，用于初始化生成器

  1. 整个程序运行细节非常重要，详细拆解如下
  • 第一步a.send(None)，是执行第一次循环，运行到yield i。注意：没有运行r = yield赋值这一步，这也是第一次只能传入None的原因
  • 第二步a.send(1)，先r = yield赋值，将send进去的1赋值给了r；然后print(r)即第一个打印出来的1；之后进入第二次循环，运行到yield i没有赋值，把yield的结果1打印出来，即第二个1
  • 最后一次a.send(1)先对r赋值，再print(r)；然后就退出了循环，没有什么可以yield的了，于是抛出异常

有了send这样的机制，我们就可以实现函数之间的来回切换执行，这是协程的基础。

廖雪峰老师网站上用这一特性完成了一个类似生产者消费者模式的例子，读者可以看看根据上面的知识能不能看懂这个例子

def consumer():
    r = ''
while True:
        n = yield r
if not n:
return
        print('[CONSUMER] Consuming %s...' % n)
        r = '200 OK'
def produce(c):
    c.send(None)
    n = 0
while n < 5:
        n = n + 1
        print('[PRODUCER] Producing %s...' % n)
        r = c.send(n)
        print('[PRODUCER] Consumer return: %s' % r)
    c.close()
c = consumer()
produce(c)
复制代码

运行结果如下

[PRODUCER] Producing 1...
[CONSUMER] Consuming 1...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 2...
[CONSUMER] Consuming 2...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 3...
[CONSUMER] Consuming 3...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 4...
[CONSUMER] Consuming 4...
[PRODUCER] Consumer return: 200 OK
[PRODUCER] Producing 5...
[CONSUMER] Consuming 5...
[PRODUCER] Consumer return: 200 OK
复制代码

上面例子看起来复杂高大上，但是如果从实现上述功能上来看，实际上是走了弯路，下面的代码可以实现和上面一样的功能，读者可以细细品味

def consumer(n):
if not n:
return
    print('[CONSUMER] Consuming %s...' % n)
return '200 OK'
def produce():
    n = 0
while n < 5:
        n = n + 1
        print('[PRODUCER] Producing %s...' % n)
        r = consumer(n)
        print('[PRODUCER] Consumer return: %s' % r)
produce()
复制代码

return yield

有时会看到return yield的用法，其实return只是起到终止函数的作用，先看下面这个函数

def myfun(total):
yield from range(total)
复制代码
>>> a = myfun(4)
>>> a
<generator object myfun at 0x000001B61CCB9CA8>
>>> for i in a:
...     print(i)
...
0
1
2
3
复制代码

这样a就是个生成器。如果用return也一样

# 不加括号不合规定
>>> def myfun(total):
...     return yield from range(total)
  File "<stdin>", line 2
return yield from range(total)
               ^
SyntaxError: invalid syntax
>>> def myfun(total):
...     return (yield from range(total))
...
>>> a = myfun(4)
>>> for i in a:
...     print(i)
...
0
1
2
3
复制代码

不过下面两个不一样

def myfun1(total):
return (yield from range(total))
yield 1 # 这个1在return后面，不会再执行
def myfun2(total):
yield from range(total)
yield 1 
复制代码

看下面运行结果

>>> a = myfun1(3)
>>> for i in a:
...     print(i)
...
0
1
2
>>> a = myfun2(3)
>>> for i in a:
...     print(i)
...
0
1
2
1
复制代码

欢迎关注我的知乎专栏

专栏主页：python编程

专栏目录：目录

版本说明：软件及包版本说明

weixin_33858485
关注
JAVA知识点全面总结8：线程
qq_44724899的博客
03-06 203
学习
js 前端 异步 async await promise resolve reject generator yield setTimeout 请求 原理 面试
cuandeqin2083的博客
03-29 580
最近在写【重拾前端】系列，下面有几个快速通道，大家自取 【重识前端】原型/原型链和继承 【重识前端】闭包与模块 【重识前端】全面攻破this 【重识前端】一次搞定JavaScript的执行机制 【重识前端】什么是BFC、IFC、GFC 和 FFC 【重识前端】深入内存世界 【重识前端】暴走的异步编程 前言 老规矩，还是先了解一下什么是异步。异步其实是一个相对比较高级的一个概念。 通常来说，程序都是顺序执行，同一时刻只会发生一件事。如果一个函数依赖于另一个函数的结果，它只能等待那个函数结束才能继续执行，从用户
Python 中 yield 的用法理解 与 send() 函数对生成器赋值
最新发布
Z2572862506的博客
01-26 331
Python 中 yield 的用法理解 与 send() 函数对生成器赋值
yield 详解
qq_37472826的博客
10-08 2646
通常的for...in...循环中，in后面是一个数组，这个数组就是一个可迭代对象，类似的还有链表，字符串，文件。它可以是mylist = [1, 2, 3]，也可以是mylist = [x*x for x in range(3)]。 它的缺陷是所有数据都在内存中，如果有海量数据的话将会非常耗内存。 生成器是可以迭代的，但只可以读取它一次。因为用的时候才生成。比如 mygenerator = (x*x for x in range(3))，注意这里用到了()，它就不是数组，而上面的例子是[]。 我理解
python中yield的用法详解——最简单，最清晰的解释
spider_man
11-29 1228
首先，如果你还没有对yield有个初步分认识，那么你先把yield看做“return”，这个是直观的，它首先是个return，普通的return是什么意思，就是在程序中返回某个值，返回之后程序就不再往下运行了。看做return之后再把它看做一个是生成器（generator）的一部分（带yield的函数才是真正的迭代器），好了，如果你对这些不明白的话，那先把yield看做return,然后直接看下面...
yield用法
anhui_andy的专栏
10-04 1175
这里写自定义目录标题模块导入发生了什么一个.py文件的用处 模块导入发生了什么 1.执行导入模块的py文件 2.产生模块.py名称空间，把模块运行中产生的名称都存放在该名称空间 3.在导入的文件中产生模块名的变量，该变量指向步骤2中产生的名称空间 一个.py文件的用处 我们对Markdown编辑器进行了一些功能拓展与语法支持，除了标准的Markdown编辑器功能，我们增加了如下几点新功能，帮助你用它写博客： 当做模块导入，name’模块名’； 当做程序执行 name=‘main’； 在这里插入代码片
Python中yield的理解与使用
热门推荐
家有代码初写成 的博客
06-28 11万+
总结：通常的for...in...循环中，in后面是一个数组，这个数组就是一个可迭代对象，类似的还有链表，字符串，文件。它可以是mylist = [1, 2, 3]，也可以是mylist = [x*x for x in range(3)]。它的缺陷是所有数据都在内存中，如果有海量数据的话将会非常耗内存。生成器是可以迭代的，但只可以读取它一次。因为用的时候才生成。比如 mygenerator = (...
Python深度解析：yield与generator的全面指南
### 总结 `yield`和生成器是Python中强大的工具，它们提供了迭代的灵活方式，并在处理大数据或持续生成数据时提高了性能。理解并熟练使用这些概念，可以极大地提升Python编程的效率和代码质量。
不错的线程总结(全面上)1
08-08
本文将全面总结Java中的线程概念，以及如何创建和管理线程。线程允许程序中的不同部分同时执行，从而提高了程序的效率和响应性。 Java提供了两种创建线程的方式： 1. **继承Thread类**：创建一个新的类，该类继承...
eslint校验规则——基于官方文档全面总结
愿得一人心&&的博客
04-09 1264
Tip：基于ESLint中文网创作 更新时间：2020年04月09日 Javascript容易出错的地方（基础错误） for-direction 强制 “for” 循环中更新子句的计数器朝着正确的方向移动 getter-return 强制 getter 函数中出现 return 语句 no-async-promise-executor 禁止使用异步函数作为 Promise executor ...
彻底理解Python中的yield
weixin_45209433的博客
09-26 9600
阅读别人的python源码时碰到了这个yield这个关键字，各种搜索终于搞懂了，在此做一下总结： 1、通常的for…in…循环中，in后面是一个数组，这个数组就是一个可迭代对象，类似的还有链表，字符串，文件。它可以是mylist = [1, 2, 3]，也可以是mylist = [x*x for x in range(3)]。 它的缺陷是所有数据都在内存中，如果有海量数据的话将会非常耗内存。 2、生成器是可以迭代的，但只可以读取它一次。因为用的时候才生成。比如 mygenerator = (x*x for
18.弄懂生成器节约内存一边计算一边生成
风中一叶
11-09 1470
  1.使用列表生成式可以方便快捷的生成列表，但是若生成一个很大的列表，势必会占用大量的内容。那我们是否能在计算的过程中，一边计算一边生成喃？答案是肯定的，我们使用生成器generator。   2.生成器的创建很简单，最简单的一种是把生成式的[]替换成() a= [x*x for x in range(10)] print(type(a)) b= (x*x for x in range(...
yield表达式, 四种形式
hunyxv的博客
03-10 3072
yield表达式, 四种形式： a. 不接受输入值或者输入值是None yield 1 b. 接受输入值 s = yield 1 c. 接受输入，但不返回数据，这样默认返回None s = yield d.既不接受输入，也不返回值，默认返回None yield 第一种：当函数调用到yield时，返回y
完美理解ruby中的yield的概念
lgx86512的专栏
05-26 1562
地址：http://www.cnitblog.com/kenlistian/archive/2008/03/16/40979.html   看到axgle 兄解释ruby中的yield的概念，形象的理解为“占位”的作用。 但是对yield还可以带着参数的概念总觉得有点不够形象，  def foo yield [1,2,3] end foo {|v| p v}以我看来更像是，比喻成一
布局中的yield
qwbtc的博客
07-14 349
在布局中，yield 标明一个区域，渲染的视图会插入这里。最简单的情况是只有一个 yield，此时渲染的整个视图都会插入这个区域： html>   head>   head>   body>    yield %>   body> html> 布局中可以标明多个区域：
C#中yield return用法分析
weixin_34185364的博客
06-10 276
这篇文章主要介绍了C#中yield return用法,对比使用yield return与不使用yield return的流程,更直观的分析了yield return的用法,需要的朋友可以参考下． 本文实例讲述了C#中yield return用法，并且对比了使用yield return与不使用yield return的情况，以便读者更好的进行理解。具体如下： yield关键字用于遍历循环中，yi...
TS错误信息列表
Jinmy的博客
07-11 9万+
错误信息列表code类型英文描述中文描述1002错误Unterminated string literal.未终止的字符串文本。1003错误Identifier expected.应为标识符。1005错误'{0}' expected.应为“{0}”。1006错误A file cannot have a reference to itself.文件不能引用自身。1009错误Trailing comm...
浅谈Python中的yield表达式
大音希声大象无形
06-25 1万+
浅谈Python协程中的yield表达式python生成器python中生成器是迭代器的一种，使用yield返回函数值。每次调用yield会暂停，而可以使用next()函数和send()函数可以恢复生成器。 这里可以参考Python函数式编程指南：生成器注意到yield是个表达式而不仅仅是个语句，所以可以使用x = yield r 这样的语法。这个知识点在协程中需要使用。协程的概念指的是在一个线程
unity协程_Unity中的Yield Return是个什么鬼？
weixin_39723010的博客
12-11 887
写完以后发现了这篇博文，看完觉得自己太浅薄了。Coroutine，你究竟干了什么？_tkokof1的专栏-CSDN博客​blog.csdn.net好多年没怎么用Unity，回过头来一看已经2020版本了，哈哈，当年最早还是用1.7版本，一帮菜鸟级码农在出租屋里写着大量if嵌套的丑陋代码，后来自己跟着一位大神开始C#+Unity之不归路，当时还是3.5版本，而自己做了个小项目的时候升到了4.3。真是...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值