python生成器代码

以下是两个Python生成器的例子：

1.使用yield关键字创建生成器函数

def my_generator(n):
    for i in range(n):
        yield i*i

# 调用生成器函数
gen = my_generator(5)

# 遍历生成器对象
for num in gen:
    print(num)

输出：

0
1
4
9
16

2.使用生成器表达式创建生成器对象

# 创建生成器对象
gen = (i*i for i in range(5))

# 遍历生成器对象
for num in gen:
    print(num)

输出：

0
1
4
9
16