迭代器，生成器

### Python 迭代器、生成器和装饰器的概念及用法 #### 1. 迭代器（Iterator） 迭代器是一种可以记住遍历位置的对象，它实现了两个方法：`__iter__()` 和 `__next__()`。`__iter__()` 方法返回迭代器对象自身，而 `__next__()` 方法返回容器的下一个元素。如果已经没有更多元素，则抛出 `StopIteration` 异常[^1]。 以下是通过实现 `__iter__()` 和 `__next__()` 方法手动创建迭代器的示例： ```python 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): value = self.data[self.index] self.index += 1 return value else: raise StopIteration # 使用自定义迭代器 my_iter = MyIterator([1, 2, 3]) for item in my_iter: print(item) ``` 此外，还可以通过 `iter()` 函数将一个可迭代对象转换为迭代器[^3]。 #### 2. 生成器（Generator） 生成器是一种特殊的迭代器，它通过函数和 `yield` 表达式来实现。当函数中包含 `yield` 关键字时，该函数不再是一个普通函数，而是一个生成器。生成器会在每次调用 `next()` 方法时暂停并保存当前状态，直到下一次调用 `next()` 时继续执行。 以下是一个简单的生成器示例： ```python def countdown(n): while n > 0: yield n n -= 1 # 创建生成器对象 generator = countdown(5) for value in generator: print(value) ``` 从 Python 3.3 开始，引入了 `yield from` 语法，用于将一个可迭代对象委托给生成器。这可以简化生成器的实现，使其更加清晰和易读[^2]。 ```python def countdown(n): while n > 0: yield n n -= 1 def my_gen(n): yield from countdown(n) yield n + 1 # 创建生成器对象 generator = my_gen(5) print(list(generator)) # [5, 4, 3, 2, 1, 6] ``` #### 3. 装饰器（Decorator） 装饰器是一种用于修改函数或方法行为的高级 Python 技术。它本质上是一个函数，接受一个函数作为参数，并返回一个新的函数。装饰器通常用于日志记录、访问控制、性能测试等场景[^3]。 以下是一个简单的装饰器示例： ```python def my_decorator(func): def wrapper(*args, **kwargs): print("Before function call") result = func(*args, **kwargs) print("After function call") return result return wrapper @my_decorator def say_hello(name): print(f"Hello, {name}!") say_hello("Alice") # 输出：Before function call\nHello, Alice!\nAfter function call ``` 装饰器可以通过堆叠多个装饰器来增强功能。例如： ```python def decorator_one(func): def wrapper(*args, **kwargs): print("Decorator One Before") result = func(*args, **kwargs) print("Decorator One After") return result return wrapper def decorator_two(func): def wrapper(*args, **kwargs): print("Decorator Two Before") result = func(*args, **kwargs) print("Decorator Two After") return result return wrapper @decorator_one @decorator_two def greet(): print("Greeting") greet() ``` #### 总结 - **迭代器** 是一种可以记住遍历位置的对象，需要实现 `__iter__()` 和 `__next__()` 方法。 - **生成器** 是一种特殊的迭代器，使用 `yield` 表达式实现，支持 `yield from` 语法以简化代码[^2]。 - **装饰器** 是一种用于修改函数或方法行为的技术，通过包装函数实现功能扩展。