生成器-代码举例：（）和yield

怎么自定义一个生成器：
两个方法：

1、小括号包裹表达式

2、函数中用yield返回

 

方法一：①小括号包裹表达式

G=(x*2 for x in range(5))

print（G）输出：<generator object ....>，3次next(G)，next(G)，next(G)后，输出0，2，4

方法二：②函数中用yield返回
每次调用next函数，会执行生成器代码，一直到yield返回一个数据。下次调用next函数，会从当前停止的yield之后开始执行。

（1）函数式生成器的简单形式

def G():
    i = 0
    yield i
    i += 1
    yield i
    i += 1
    yield i
g1=G()
print(next(g1))
print(next(g1))
print(next(g1))

说明：

使用了yield后，就不再是函数，而是生成器。
yield的作用有2点：一是记录当前运行状态（断点），二是返回值，类似return。
next()函数让生成器从断点出继续执行，即唤醒生成器。

（2）函数式生成器的循环形式

def G2():
    i = 0
    while i < 5:
        yield i  # 下次调用next函数，会从yield之后开始执行。
        i += 1


g1 = G2()
print(next(g1))  # 每次调用next函数，会执行生成器的代码，一直到yield返回一个数据。
print(next(g1))
print(next(g1))

输出：

0

1

2

 

（3）函数式迭代器 - 斐波拉契数列的生成器

def fib_generator(n):
    '''斐波那契数列生成器

    :param n: 可遍历（迭代）的次数
    :return: yield
    '''
    current = 0  # 标记迭代器位置
    # 初始值
    num1 = 0
    num2 = 1

    while current < n:
        num = num1
        num1, num2 = num2, num1 + num2
        current += 1
        yield num


def main():
    f = fib_generator(10)
    for x in f:
        print(x, end=", ")


if __name__ == '__main__':
    main()

输出：

0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 

 

转载于:https://www.cnblogs.com/andy9468/p/10406316.html