本文实例讲述了Python使用迭代器捕获Generator返回值的方法。分享给大家供大家参考,具体如下:
用for循环调用generator时,发现拿不到generator的return语句的返回值。如果想要拿到返回值,必须捕获StopIteration错误,返回值包含在StopIteration的value中:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#!/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
|
输出:
1
2
3
4
5
6
7
|
g = 1
g = 1
g = 2
g = 3
g = 5
g = 8
Generrator return value: done
|
希望本文所述对大家Python程序设计有所帮助。