您的位置:首页 > 编程语言 > Python开发

【python】理解迭代器及yield

2014-03-30 21:10 274 查看
虽然很早之前就接触yield这个词了,却一直是一知半解。趁现在有时间,把它研究一通再说。

含有yield的函数说明它是一个生成器,而不是普通的函数。当程序运行到yield这一行时,该函数会返回值,并保存当前域的所有变量状态;等到该函数下一次被调用时,会从上一次中断的地方开始执行,一直遇到下一个yield, 程序返回值, 并在此保存当前状态; 如此反复,直到函数正常执行完成。

我一开始还想不明白调用者与生成器之间的函数堆栈是怎么做到的,后来才大悟原来是用到了'协程'这个原理。协程可视为微线程,下面会结合例子来说明一下yield及协程的运行过程。假设定义了test方法:

def test(len):
    i = 0
    while i < len :
	    yield i
	    i += 1
我们来调用它看看输出:

>>> for i in test(5):

print i

输出:

0

1

2

3

4

这场景是不是很类似 for i in xrange(len); 是的, xrange就是这么干的。 for .. in 的操作实际上是调用了生成器的next()方法,以上的调用过程可以等价为:

f = test(5)
print f.next()
print f.next()
print f.next()
print f.next()
print f.next()
输出结果与上次输出一致。

另外,在这次调用过程中,协程被创建了一次, 被唤醒了5次(通过next),被挂起了5次(通过yield), 最后协程退出并销毁。 大概就这些点了,有更深的理解再做补充。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: