python generator 示例代码

以下是一个简单的 Python 生成器示例代码：

def my_generator(n):
    i = 0
    while i < n:
        yield i
        i += 1

# 使用 for 循环输出生成器中的值
for num in my_generator(5):
    print(num)

在上面的示例中，`my_generator` 是一个生成器函数，它接受一个整数 `n` 作为参数，并在 `while` 循环中使用 `yield` 语句生成从 0 到 `n-1` 的整数。使用 `yield` 而不是 `return` 语句使得该函数返回一个生成器对象，而不是一个列表或其他类型的数据结构。

在主程序中，使用 `for` 循环遍历生成器对象并输出其中的值。由于生成器对象只会在需要时生成下一个值，因此该示例代码非常高效，并且不会占用大量内存。