PythonL-Chapter_03_迭代器与生成器

在Python中，迭代器（Iterator）和生成器（Generator）是用于处理可迭代对象的重要工具。它们提供了高效的数据处理方式，特别是在处理大数据集时。

迭代器

迭代器是一个实现了__iter__()和__next__()方法的对象。iter()方法返回迭代器对象本身，而__next__()方法返回序列中的下一个值。如果没有更多的元素，next()方法会引发StopIteration异常。

示例

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_iterator = MyIterator([1, 2, 3, 4, 5])
for item in my_iterator:
    print(item)  # 输出: 1 2 3 4 5

1
2
3
4
5

生成器

生成器是一种特殊的迭代器，使用yield关键字来生成值。生成器函数在调用时返回一个生成器对象，该对象可以用于迭代。生成器函数在每次调用yield时会暂停执行，并返回一个值，下次调用时会从上次暂停的地方继续执行。

示例

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

# 使用生成器
my_gen = my_generator([1, 2, 3, 4, 5])
for item in my_gen:
    print(item)  # 输出: 1 2 3 4 5

1
2
3
4
5