这篇文章主要介绍了python使用迭代器捕获generator返回值的方法,结合具体实例形式分析了python迭代器获取生成器返回值的相关操作技巧,需要的朋友可以参考下
本文实例讲述了python使用迭代器捕获generator返回值的方法。分享给大家供大家参考，具体如下：
用for循环调用generator时，发现拿不到generator的return语句的返回值。如果想要拿到返回值，必须捕获stopiteration错误，返回值包含在stopiteration的value中：
#!/usr/bin/env python
# -*- coding: utf-8 -*-
def fib(max):
n, a, b = 0, 0, 1
while n < max:
yield b
a, b = b, a + b
n = n + 1
return 'done'
# 捕获generator的返回值
g = fib(6)
while true:
try:
x=next(g)
print('g=',x)
except stopiteration as e:
print('generrator return value:', e.value)
break
输出：
g= 1
g= 1
g= 2
g= 3
g= 5
g= 8
generrator return value: done
以上就是python使用迭代器捕获generator返回值的方法的详细内容。