Python中的可迭代对象、迭代器、生成器和装饰器-CSDN博客

本文链接：https://blog.csdn.net/2302_78656999/article/details/141563328

Python是一种功能强大且灵活的编程语言，它提供了多种高级特性来简化代码和提高效率。本文将深入探讨Python中的可迭代对象（Iterable）、迭代器（Iterator）、生成器（Generator）和装饰器（Decorator），帮助你更好地理解和应用这些概念。

1. 可迭代对象（Iterable）

可迭代对象是可以被迭代的对象，例如列表、元组、字符串、字典等。在Python中，一个对象如果实现了 __iter__()方法或 __getitem__()方法，那么它就是可迭代的。

# 列表是可迭代对象
my_list = [1, 2, 3]
for item in my_list:
    print(item)

2. 迭代器（Iterator）

迭代器是一个实现了 __iter__()和 __next__()方法的对象。迭代器可以记住遍历的位置，并且每次调用 __next__()方法时返回下一个值，直到没有更多元素时抛出 StopIteration异常。

class MyIterator:
    # 定义类的初始化方法 __init__，它接受两个参数：self（代表类的实例）和 data（要迭代的数据）
    def __init__(self, data):
        self.data = data
        # 初始化一个实例变量 self.index，并将其设置为 0，表示从数据的第一个元素开始迭代
        self.index = 0

    def __iter__(self):
        # 定义 __iter__ 方法，这个方法是迭代器协议的一部分。它返回迭代器对象本身，使得 MyIterator 实例可以被用于 for 循环
        return self

    def __next__(self):
        # 定义 __next__ 方法，这个方法也是迭代器协议的一部分。它负责返回序列中的下一个值，并在没有更多值时引发 StopIteration 异常。
        if self.index >= len(self.data):
            raise StopIteration
        value = self.data[self.index]
        self.index += 1
        return value

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

3. 生成器（Generator）

生成器是一种特殊的迭代器，它使用 yield关键字来生成值。生成器函数在调用时返回一个生成器对象，每次调用 next()函数时，生成器函数会执行到下一个 yield语句处，并返回生成的值。

def my_generator():
    yield 1
    yield 2
    yield 3

gen = my_generator()
for item in gen:
    print(item)

生成器的一个常见应用是处理大数据集，因为它可以按需生成数据，而不是一次性加载所有数据到内存中。

def fibonacci(n):
    # 定义一个名为 fibonacci 的函数，它接受一个参数 n，表示要生成多少个斐波那契数列的元素
    a, b = 0, 1
    # 初始化两个变量 a 和 b，分别赋值为 0 和 1，这是斐波那契数列的前两个数字
    for _ in range(n):
        # 使用一个 for 循环，循环 n 次。这里的 _ 是一个占位符，表示我们不关心循环的索引值
        yield a
        # 使用 yield 关键字生成当前的斐波那契数 a，并暂停函数的执行，将 a 返回给调用者。当函数再次被调用时，会从这里继续执行
        a, b = b, a + b
        # 更新 a 和 b 的值。a 变为原来的 b，b 变为原来的 a 和 b 的和，这是斐波那契数列的递推公式

for num in fibonacci(10):
    # 调用 fibonacci 函数，生成前 10 个斐波那契数。这里的 for 循环会自动处理 yield 返回的值，每次循环获取一个生成的斐波那契数
    print(num)
    # 在每次循环中，打印当前获取的斐波那契数 num

4. 装饰器（Decorator）

装饰器是一种用于修改函数或方法行为的高级技术。装饰器本质上是一个函数，它接受一个函数作为参数，并返回一个新的函数。

def my_decorator(func):
    def wrapper(*args, **kwargs):
        print("函数被调用前发生了")
        result = func(*args, **kwargs)
        print("函数被调用后发生了")
        return result
    return wrapper

@my_decorator
def say_hello(name):
    print(f"Hello, {name}!")

say_hello("Alice")

装饰器可以用于日志记录、性能监控、权限检查等多种场景。

import time

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"Function {func.__name__} took {end_time - start_time} seconds to run.")
        return result
    return wrapper

@timer
def slow_function():
    time.sleep(2)

slow_function()

总结

本文介绍了Python中的可迭代对象、迭代器、生成器和装饰器。可迭代对象是可以被迭代的对象，迭代器是实现了 __iter__()和 __next__()方法的对象，生成器是使用 yield关键字的特殊迭代器，装饰器是用于修改函数或方法行为的高级技术。希望这篇文章能帮助你更好地理解和应用这些概念，并在实际编程中发挥作用。