Python迭代器捕获Generator返回值 Python使用迭代器捕获Generator返回值的方法

时间:2021-03-24 FrankYou 人气:5

想了解Python使用迭代器捕获Generator返回值的方法的相关内容吗，FrankYou在本文为您仔细讲解Python迭代器捕获Generator返回值的相关知识和一些Code实例，欢迎阅读和指正，我们先划重点：Python,迭代器,捕获,Generator,返回值，下面大家一起来学习吧。

本文实例讲述了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程序设计有所帮助。

加载全部内容