python中的yield使用方法

今天在看其他同事的代码时,发现一个没使用过的python关键字 :yield

先问了一下同事,听他说了几句,有个模糊的印象,仅仅是模糊而已。于是自己去搜搜资料看。看了半天,逐渐清晰了。不过在工作机制以及应用上还是有点迷茫。嗯,先把初始接触的印象记下来吧。

yield 简单说来就是一个生成器(generator)。生成器是这样一个函数:它记住上一次返回时在函数体中的位置。对生成器函数的第二次(或第 n 次)调用跳转至该函数中间,而上次调用的所有局部变量都保持不变。

你看到某个函数包含了yield,这意味着这个函数已经是一个generator,它的执行会和其他普通的函数有很多不同。

可能看到这里还是迷迷糊糊,先看一些实例吧:

代码如下:

def test( data_list ):

for x in data_list:

yield x + 1

data = [1,2,3,4]

for y in test( data ):

print y

则输出结果为:

2 3 4 5

另外一种用法:

handle = test(data)

handle.next() 输出 2

handle.next() 输出 3

handle.next() 输出 4

handle.next() 输出 5

handle.next() 则会报错

这只是 yield 的初步印象,接下来慢慢摸索吧。