python特性（六）：yield与迭代器生成器

转自 http://blog.csdn.net/hedan2013/article/details/56017612

 

 

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

 

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

 

例如，看下面的例子。

 

 

def MyGenerator():
    yield 1

gen = 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异常。

 

 

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