首页 > 编程 > Python > 正文

python特性(六):yield与迭代器生成器

2019-11-08 01:44:24
字体:
来源:转载
供稿:网友

迭代器是非常高效的类型,无论是从时间复杂度,还是从空间复杂度。而实现迭代器的代码虽然简单,却也繁琐。为此,python定义了一个yield关键字,专门用来构造迭代器。yield有生成,产生的意思。

yield的功能和return非常类似,它们都只能在方法中使用。不同的是,包含yield语句的方法被称为生成器方法。当调用生成器方法时,会返回一个生成器对象。

例如,看下面的例子。

def MyGenerator():    yield 1gen = MyGenerator()PRint gen

输出结果为

<generator object MyGenerator at 0x0000000001D9DD80>

当调用生成器对象的next方法时,会执行生成器方法中的代码,直至遇到yield语句时,方法的执行过程会被挂起。同时,方法运行的上下文环境会被保存。而next方法的返回值就是yield关键字后面表达式的返回值。

例如,下面代码

print gen.next()执行结果为

1当我们继续调用next方法时,从上一次挂起的地方开始,继续执行后面的代码。直至遇到下一个yield语句。当方法执行完毕,依然没有遇到yield语句,抛出StopIteration异常。

例如

def MyGenerator():    yield 1    yield 'a'gen = MyGenerator()print gen.next()print gen.next()print gen.next()上面代码中第1次调用next方法,执行语句yield 1。第2次调用next方法,执行语句yield 'a'。第3次调用next方法时,在方法退出前都没有遇到yield语句,因此抛出StopIteration异常。

上面介绍的生成器方法的工作机理。在后面的博文中,会逐步介绍生成器方法的一些经典应用。


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表