详解Python生成器函数和生成器对象的原理和用法

包含yield语句的函数可以用来创建生成器对象,这样的函数也称生成器函数。yield语句与return语句的作用相似,都是用来从函数中返回值。与return语句不同的是,return语句一旦执行会立刻结束函数的运行,而每次执行到yield语句并返回一个值之后会暂停后面代码的执行,下次通过生成器对象的__next__()方法、内置函数next()、for循环遍历生成器对象元素或其他方式显式“索要”数据时继续执行。生成器具有惰性求值的特点,适合大数据处理。下面的代码演示了如何使用生成器来生成斐波那契数列:

>>> def f():

a, b = 1, 1 #序列解包,同时为多个元素赋值

while True:

yield a #返回一个值,暂停执行,需要时再产生一个新元素

a, b = b, a+b #序列解包,继续生成新元素

>>> a = f() #创建生成器对象

>>> for i in range(10): #斐波那契数列中前10个元素

print(a.__next__(), end=' ')

1 1 2 3 5 8 13 21 34 55

>>> for i in f(): #斐波那契数列中第一个大于100的元素

if i > 100:

print(i, end=' ')

break

144

>>> a = f()

>>> next(a) #使用内置函数next()获取生成器对象中的元素

1

>>> next(a) #每次索取新元素时,由yield语句生成

1

>>> a.__next__() #也可以调用生成器对象的__next__()方法

2

>>> a.__next__()

3

Python标准库itertools提供了一个count(start, step)函数,用来连续不断地生成无穷个数,这些数中的第一个数是start,相邻两个数的差是step。下面的代码使用生成器模拟了标准库itertools中的count()函数。

>>> def count(start, step):

num = start

while True:

yield num

num += step

>>> x = count(3, 5)

>>> for i in range(10):

print(next(x), end=' ')

3 8 13 18 23 28 33 38 43 48

>>> for i in range(10):

print(next(x), end=' ')

53 58 63 68 73 78 83 88 93 98

原文发布于微信公众号 - Python小屋(Python_xiaowu)

原文发表时间:2016-11-19

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏null的专栏

挑战数据结构与算法面试题——统计上排数在下排出现的次数

题目来源“数据结构与算法面试题80道”。在此给出我的解法,如你有更好的解法,欢迎留言。 ? 分析: 本题应该是一个确定的问题,即上排的是个数是题目中给定的...

3246
来自专栏Java帮帮-微信公众号-技术文章全总结

第十三天 面向对象-final static 匿名对象内部类包代码块【悟空教程】

1464
来自专栏Web行业观察

一张图撸明白prototype原型链

expression instanceof class expression和class都是必选项。

3274
来自专栏java工会

Java基础第一阶段知识点,招实习的面试官都在问这些

2229
来自专栏ml

javaSE基础之基本细节注解

  1.  对于多行注释而言,不能进行嵌套注释.....! /* dada /* d adasdas */...

2694
来自专栏青玉伏案

窥探Swift之新添数据类型元组与可选值

  今天的博客中就总结一下关于Swift中相对Objc新添加的两个数据类型:元组(Tuple)和可选值类型(Optional)。上面这两个类型是Swift独有的...

2015
来自专栏java工会

Java基础第一阶段知识点,招实习的面试官都在问这些

a) 答:Java源文件被编译成字节码的形式,无论在什么系统环境下,只要有java虚

1361
来自专栏开发与安全

从零开始学C++之从C到C++(一):const与#define、结构体对齐、函数重载name mangling、new/delete 等

一、bool 类型 逻辑型也称布尔型,其取值为true(逻辑真)和false(逻辑假),存储字节数在不同编译系统中可能有所不同,VC++中为1个字节。 声明方式...

1940
来自专栏程序员互动联盟

【专业技术】你必须注意的11个C++要点

下面的这些要点是对所有的C++程序员都适用的。我之所以说它们是最重要的,是因为这些要点中提到的是你通常在C++书中或网站上无法找到的。如:指向成员的指针,这是许...

2765
来自专栏CVer

排序算法 | 冒泡排序(含C++/Python代码实现)

排序算法,就是如何使得记录按照要求排列的方法。排序算法在很多领域得到相当地重视,尤其是在大量数据的处理方面。排序算法有很多,本文将介绍最经典的排序算法:冒泡排序...

1322

扫码关注云+社区

领取腾讯云代金券