Python 3 中生成器函数yield表达式的使用

生成器函数或生成器方法中包含了一个yield表达式。调用生成器函数时,会返回一个迭代子,值从迭代子中每次提取一个(通过调用其__next__()方法)。每次调用__next__()时,生成器函数的yield表达式的值(如果未指定就是None)都会返回。如果生成器函数结束或执行一个return,就产生StopIteration异常。

上面的说法有点官方,下边是一些个人的理解:

1、当一个函数中包含有yield表达式时,在调用这个调用这个函数时,函数的代码并不会运行,而是返回一个迭代子,或者说是一个可用于迭代的容器(相当于一个数组的序列数)。如下面这个例子,当调用a()时并没有运行代码

>>> def a():

            print('a')

            yield 'x'

>>> a()

<generator object a at 0x0000027BBFB380F8>

>>>

2、要想使这个函数运行起来,就需要一个迭代器来读取这个容器内的内容,比如说for语句,如下:

>>> for i in a():
i a
'x'
>>>

3、一个yield表达式只会产生“一个“迭代子,再解释一下,这个迭代子就同于序列[‘a’,’b’,’c’]中‘a’,’b’,’c’的位置参数0,1,2。如果有多个迭代子,当调用迭代器时,就会有多次迭代:

def b():

    print('a')

    yield 1

    print('b')

    yield 3

for i in b():

    print(i)

a

1

b

3

>>>

例子中的for循环执行了两次,第一次迭代的是yield 1产生的迭代子(也可以简单的把它理解为序列的第一个序列号0),返回了结果1和yield 表达式前边的print(‘a’) 的结果,第二次迭代的是yield 3表达式产生的迭代子,结果为yield 3表达式的返回值3和它前边的语句print(‘b’)的结果。

4、yield在一定程度上也有return的作用——都会返回一个值(即上面函数的‘x’)但yield不会终止函数,而是使函数暂停,直到所有的迭代子都被使用后,就会产生一个StopIteration异常,从而终止函数

5、用iter函数来查看可迭代结构的具体过程。

用a=iter(iterable结构)可以获取一个迭代子,在每次循环时可用next(a)方法获取下一个数据项,当结尾时会产生一个StopIteration异常。

函数b()相当于一个包含了两个迭代子的容器;

>>> c=iter(b())

>>> next(c)

a

1

>>> next(c)

b

3

>>> next(c)

Traceback (most recent call last):

File "<pyshell#6>", line 1, in <module>

next(c)

StopIteration

>>>

第一次和第二次迭代分别返回两个迭代子的结果,当所有的迭代子都已使用时,第三次迭代就产生了一个StopIteration。

6、yield返回的就是迭代子的值,yield可以将任意内容设置为迭代子。大多数情况下我们理解的迭代子都是0,1,2,3,4...这样的自然位置迭代,而yield就把它设置为任意的值,a()函数生成的迭代子为 ‘x’,b()函数生成的迭代子为1,3。

上一篇:codevs2806 红与黑


下一篇:【刷题】BZOJ 2049 [Sdoi2008]Cave 洞穴勘测