深入学习python的yield和generator


Posted in Python onMarch 10, 2016

前言
没有用过的东西,没有深刻理解的东西很难说自己会,而且被别人一问必然破绽百出。虽然之前有接触过python协程的概念,但是只是走马观花,这两天的一次交谈中,别人问到了协程,顿时语塞,死活想不起来曾经看过的东西,之后突然想到了yield,但为时已晚,只能说概念不清,所以本篇先缕缕python的生成器和yield关键字。

什么是生成器
1、生成器是一个特殊的程序,可以被用作控制循环的迭代行为
2、生成器类似于返回值为数组的一个函数,这个函数可以接收参数,可以被调用,但是,不同于一般的函数会一次性返回包含了所有数值的数组,生成器一次只产生一个值,这样消耗的内粗数量大大减少,而且允许调用函数可以很快的开始处理前几个返回值。因此,生成器看起来像一个函数但是表现的却像一个迭代器。
python中的生成器
python提供了两种基本的方式。

1)、生成器函数:也是用def来定义,利用关键字yield一次返回一个结果,阻塞,重新开始
2)、生成器表达式:返回一个对象,这个对象只有在需要的时候才产生结果
下面详细讲解。

1、生成器函数
为什么叫生成器函数?因为他随着时间的推移生成了一个数值队列。一般的函数在执行完毕之后会返回一个值然后退出,但是生成器函数会自动挂起,然后重新拾起继续执行,他会利用yield关键字关起函数,给调用者返回一个值,同时保留了当前的足够多的状态,可以使函数继续执行。生成器和迭代协议是密切相关的,可迭代的对象都有一个__next()__成员方法,这个方法要么返回迭代的下一项,要么引起异常结束迭代。
为了支持迭代协议,拥有yield语句的函数被编译为生成器,这类函数被调用时返回一个生成器对象,返回的对象支持迭代接口,即成员方法__next()__继续从中断处执行执行。
看下面的例子:

# codes
def create_counter(n):
 print "create counter"
 while True:
  yield n
  print 'increment n'
  n += 1

cnt = create_counter(2)
print cnt
print next(cnt)
print next(cnt)

# output
<generator object create_counter at 0x0000000001D141B0>
create counter
2
increment n
3

分析一下这个例子:

  • 在create_counter函数中出现了关键字yield,预示着这个函数每次只产生一个结果值,这个函数返回一个生成器(通过第一行输出可以看出来),用来产生连续的n值
  • 在创造生成器实例的时候,只需要像普通函数一样调用就可以,但是这个调用却不会执行这个函数,这个可以通过输出看出来
  • next()函数将生成器对象作为自己的参数,在第一次调用的时候,他执行了create_counter()函数到yield语句,返回产生的值2
  • 我们重复的调用next()函数,每次他都会从上次被挂起的地方开始执行,直到再次遇到了yield关键字

为了更加深刻的理解,我们再举一个例子。

#coding
def cube(n):
 for i in range(n):
  yield i ** 3

for i in cube(5):
 print i

#output
0
1
8
27
64

所以从理解函数的角度出发我们可以将yield类比为return,但是功能确实完全不同,在for循环中,会自动遵循迭代规则,每次调用next()函数,所以上面的结果不难理解。

2、生成器表达式
生成器表达式来自于迭代和列表解析的组合,关于列表解析的概念和用法可以参见我之前的博客,生成器表达式和列表解析类似,但是他使用尖括号而不是方括号括起来的。如下代码:

>>> # 列表解析生成列表
>>> [ x ** 3 for x in range(5)]
[0, 1, 8, 27, 64]
>>> 
>>> # 生成器表达式
>>> (x ** 3 for x in range(5))
<generator object <genexpr> at 0x000000000315F678>
>>> # 两者之间转换
>>> list(x ** 3 for x in range(5))
[0, 1, 8, 27, 64]

就操作而言,生成器表如果使用大量的next()函数会显得十分不方便,for循环会自动出发next函数,所以可以按下面方式使用:

>>> for n in (x ** 3 for x in range(5)):
 print('%s, %s' % (n, n * n))

 
0, 0
1, 1
8, 64
27, 729
64, 4096
>>>

两者比较
一个迭代既可以被写成生成器函数,也可以被协程生成器表达式,均支持自动和手动迭代。而且这些生成器只支持一个active迭代,也就是说生成器的迭代器就是生成器本身。

总结
想起了初中时候老师经常说的,眼观千遍,不如手动一遍。

Python 相关文章推荐
python多线程编程中的join函数使用心得
Sep 02 Python
python 中的int()函数怎么用
Oct 17 Python
恢复百度云盘本地误删的文件脚本(简单方法)
Oct 21 Python
numpy中以文本的方式存储以及读取数据方法
Jun 04 Python
Python Learning 列表的更多操作及示例代码
Aug 22 Python
python中ImageTk.PhotoImage()不显示图片却不报错问题解决
Dec 06 Python
Python实现钉钉订阅消息功能
Jan 14 Python
详解Python修复遥感影像条带的两种方式
Feb 23 Python
Python新手如何理解循环加载模块
May 29 Python
详解KMP算法以及python如何实现
Sep 18 Python
Python Http请求json解析库用法解析
Nov 28 Python
python模块与C和C++动态库相互调用实现过程示例
Nov 02 Python
Python中random模块生成随机数详解
Mar 10 #Python
python生成器表达式和列表解析
Mar 10 #Python
python迭代器与生成器详解
Mar 10 #Python
Python装饰器基础详解
Mar 09 #Python
Python求算数平方根和约数的方法汇总
Mar 09 #Python
Python实现Linux命令xxd -i功能
Mar 06 #Python
基于Python实现一个简单的银行转账操作
Mar 06 #Python
You might like
Yii把CGridView文本框换成下拉框的方法
2014/12/03 PHP
php实现基于微信公众平台开发SDK(demo)扩展的方法
2014/12/22 PHP
yii2带搜索功能的下拉框实例详解
2016/05/12 PHP
jQuery下的几个你可能没用过的功能
2010/08/29 Javascript
jQuery焦点控制图层展示延迟隐藏的方法
2015/03/09 Javascript
JavaScript获取网页中第一个图片id的方法
2015/04/03 Javascript
Bootstrap 附加导航(Affix)插件实例详解
2016/06/01 Javascript
基于JS代码实现实时显示系统时间
2016/06/16 Javascript
JS数组去掉重复数据只保留一条的实现代码
2016/08/11 Javascript
js 获取站点应用名的简单实例
2016/08/18 Javascript
jquery 实现复选框的全选操作实例代码
2017/01/24 Javascript
bootstrap table表格插件使用详解
2017/05/08 Javascript
在ABP框架中使用BootstrapTable组件的方法
2017/07/31 Javascript
vue中使用vue-router切换页面时滚动条自动滚动到顶部的方法
2017/11/28 Javascript
vue实现点击展开点击收起效果
2018/04/27 Javascript
Vue实现底部侧边工具栏的实例代码
2018/09/03 Javascript
vue spa应用中的路由缓存问题与解决方案
2019/05/31 Javascript
在vue中动态添加class类进行显示隐藏实例
2019/11/09 Javascript
jQuery+Ajax+js实现请求json格式数据并渲染到html页面操作示例
2020/06/02 jQuery
解决qrcode.js生成二维码时必须定义一个空div的问题
2020/07/09 Javascript
Python操作MySQL模拟银行转账
2018/03/12 Python
Python简单实现两个任意字符串乘积的方法示例
2018/04/12 Python
Python3 安装PyQt5及exe打包图文教程
2019/01/08 Python
对DataFrame数据中的重复行,利用groupby累加合并的方法详解
2019/01/30 Python
Python实现多进程的四种方式
2019/02/22 Python
selenium python 实现基本自动化测试的示例代码
2019/02/25 Python
Python笔记之工厂模式
2019/11/20 Python
基于python-pptx库中文文档及使用详解
2020/02/14 Python
英国工具中心:UK Tool Centre
2017/07/10 全球购物
世界上最大的高分辨率在线图片库:Alamy
2018/07/07 全球购物
一套比较完整的软件测试人员面试题
2012/05/13 面试题
学校节能减排倡议书
2014/05/16 职场文书
元旦晚会活动总结
2014/07/09 职场文书
2014年班组长工作总结
2014/11/20 职场文书
物业保洁员岗位职责
2015/02/13 职场文书
2015年学生会工作总结范文
2015/03/31 职场文书