Python的yield关键字

http://blog.youkuaiyun.com/tossgoer/archive/2010/08/18/5822303.aspx

忽然得知Python有个叫yield的关键字，好奇之下去查了查，花了点时间基本弄明白了，故写在此备忘，顺便充充字数。

yield关键字用来定义生成器（Generator），其具体功能是可以当return使用，从函数里返回一个值，不同之处是用yield返回之后，可以让函数从上回yield返回的地点继续执行。也就是说，yield返回函数，交给调用者一个返回值，然后再“瞬移”回去，让函数继续运行， 直到吓一跳yield语句再返回一个新的值。

使用yield返回后，调用者实际得到的是一个迭代器对象，迭代器的值就是返回值，而调用该迭代器的next()方法会导致该函数恢复yield语句的执行环境继续往下跑，直到遇到下一个yield为止，如果遇不到yield，就会抛出异常表示迭代结束。

看一个例子：

>>> def test_yield():
...     yield 1
...     yield 2
...     yield (1,2)
...
>>> a = test_yield()
>>> a.next()
1
>>> a.next()
2
>>> a.next()
(1, 2)
>>> a.next()
Traceback (most recent call last):
  File "<stdin>", line 1, in ?
StopIteration

每次调用a.next()，都会得到下一个yield返回的值，最后一次调用会抛出StopIteration结束异常。

暂时没想到这个东西有什么用，也许就是可以省一些代码。

一个有趣的例子，用来生成一个序列的全排列：

def

 perm(

items, n=None

)

:

    if

 n is

 None

:

        n = len

(

items)

 

    for

 i in

 range

(

len

(

items)

)

:

        v = items[

i:i+1]

 

        if

 n == 1:

            yield

 v

        else

:

            rest = items[

:i]

 + items[

i+1:]

 

            for

 p in

 perm(

rest, n-1)

:

                yield

 v + p

 

 

的确是方便一些，不过我还是觉得都放到一个数组里更直观...

本文来自优快云博客，转载请标明出处：http://blog.youkuaiyun.com/tossgoer/archive/2010/08/18/5822303.aspx