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

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

时间：2020-09-28 09:47 来源/作者：FrankYou

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

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

相关文章

热门资讯