python yield 廖雪峰_python yield 与 yield from转

1、yield使用

1)函数中使用yield，可以使函数变成生成器。一个函数如果是生成一个数组，就必须把数据存储在内存中，如果使用生成器，则在调用的时候才生成数据，可以节省内存。

2)生成器方法调用时，不会立即执行。需要调用next()或者使用for循环来执行。使用for循环不需要自己捕获StopIteration异常。使用next()方法，当生产器方法执行结束会抛出StopIteration异常(只要不是使用yield返回数据，都会抛出StopIteration异常)。

示例：

def fib(max):

n,a,b = 0,0,1

while n < max:

yield b

a, b = b, a+b

n = n + 1

return 'done'

n = fib(10)

for n1 in n:

print(n1)

3)yield不仅可以返回值，也可以接收值。下面面示例为生产消费模式。生产者生产一条记录，消费者消费一条记录。

4)调用生成器send方法传递数据时，必须先调用next(c)或者c.send(None)方法，执行到yield语句，等待接收数据。否则会报错。

以下代码为廖雪峰网站的示例https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/001432090171191d05dae6e129940518d1d6cf6eeaaa969000

def consumer():

r = ''

while True:

n = yield r

if not n:

return

print('[CONSUMER] Consuming %s...' % n)

r = '200 OK'

def produce(c):

c.send(None) # 和next方法一样 获取下一个值，必须先使用None参数调用一次， 执行到yield

n = 0

while n < 5:

n = n + 1

print('[PRODUCER] Producing %s...' % n)

r = c.send(n) # 先发送值给yield语句，再执行到yield语句时返回

print('[PRODUCER] Consumer return:%s' % r)

c.close()

c = consumer()

produce(c)

2、yield from的使用

1)为了让生成器(带yield函数)，能简易的在其他函数中直接调用，就产生了yield from。

2)以下代码，htest为生成器，itest通过yield from 直接调用htest。这样itest也变成了一个生成器。创建itest实例不断的去获取数据，当生成器执行结束时，会抛出StopIteration异常。那这个异常是htest抛出的，还是itest抛出的。通过捕获异常，会发现其实是itest抛出异常，htest并不会抛出StopIteration异常。

3)yield from 也可以返回值，通过变量接收。变量接收的值，即htest使用return返回的值。示例代码中，当i==3时，会直接使用return返回，这时val的值就是100；因为htest函数中不是使用yield返回值，所以itest会继续执行print(val)语句。itest代码执行完，然而并没有使用yield返回数据(htest中没有，itest中也没有)，所以马上会抛出StopIteration异常)(如果在itest函数最后使用yield返回一个数据，就不会抛出异常)。

def htest():

i = 1

while i < 4:

n = yield i

if i == 3:

return 100

i += 1

def itest():

val = yield from htest()

print(val)

t = itest()

t.send(None)

j = 0

while j < 3:

j += 1

try:

t.send(j)

except StopIteration as e:

print('异常了')