生成器-代码举例:()和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