生成器和迭代器

生成器和迭代器是Python中用于处理可迭代对象的功能。

1. 生成器（Generator）: 生成器是一种特殊的迭代器，可以逐个地产生值而不是一次性生成所有值。生成器使用yield语句来返回一个值，并在记录位置的状态下挂起函数的执行，等待下一次调用继续执行。

示例代码：

def my_generator():
    yield 1
    yield 2
    yield 3

gen = my_generator()
print(next(gen))  # 输出 1
print(next(gen))  # 输出 2
print(next(gen))  # 输出 3

 2. 迭代器（Iterator）: 迭代器是一种实现了迭代协议（Iterator Protocol）的对象，可以用于遍历可迭代对象中的元素。迭代器必须实现__iter__()和__next__()方法。iter()方法返回迭代器对象本身，next()方法返回可迭代对象中的下一个值。

示例代码：

class MyIterator:
    def __init__(self, max_num):
        self.max_num = max_num
        self.curr_num = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.curr_num < self.max_num:
            self.curr_num += 1
            return self.curr_num
        else:
            raise StopIteration

iterator = MyIterator(3)
for num in iterator:
    print(num)  # 输出 1, 2, 3

通过生成器和迭代器，我们可以遍历大数据集而不需要一次性加载所有数据，从而节省内存空间。此外，生成器和迭代器还可以用于实现惰性计算和无限序列的生成。