生成器和迭代器的区别

生成器和迭代器的区别

迭代器和生成器都是用于处理可迭代对象的机制，但它们之间有一些区别。

迭代器是一个实现了迭代协议的对象，通过定义__iter__()和__next__()方法来实现。迭代器可以逐个返回元素，只能往前遍历，不支持索引访问。使用迭代器的好处是可以节省内存空间，因为它只在需要时生成并返回值。

生成器是一种特殊的迭代器，使用更简单。生成器是通过函数中的yield语句来实现的，它会暂停函数执行并返回一个值给调用者，保留函数的状态。当再次调用生成器时，会从上次离开的地方继续执行。生成器可以看作是一个函数的迭代器。

迭代器的代码示例

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index >= len(self.data):
            raise StopIteration
        value = self.data[self.index]
        self.index += 1
        return value

my_list = [1, 2, 3, 4, 5]
my_iterator = MyIterator(my_list)
for item in my_iterator:
    print(item)

生成器的代码示例

def my_generator(data):
    for item in data:
        yield item

my_list = [1, 2, 3, 4, 5]
for item in my_generator(my_list):
    print(item)