生成器-代码举例：（）和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))

输出：

（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,

巴特西

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

最新文章

热门文章