python的yield关键字...
时间:2010-08-17 来源:tossgoer
忽然得知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
的确是方便一些,不过我还是觉得都放到一个数组里更直观...