大家好,我是阿萨。今天看Python的装饰器和生成器。
1. 装饰器 (Decorator)
装饰器是一种在不修改原函数代码的情况下,给函数增加新功能的方法。装饰器是一个接受函数作为参数并返回一个新函数的高阶函数。在Python中,装饰器使用 `@decorator` 语法糖进行声明。
示例:计时装饰器
下面是一个简单的用于计算函数运行时间的装饰器:
```python import time def timer_decorator(func): def wrapper(*args, **kwargs): start_time = time.time() result = func(*args, **kwargs) end_time = time.time() print(f"{func.__name__} 耗时 {end_time - start_time} 秒") return result return wrapper @timer_decorator def example_function(n): total = 0 for i in range(n): total += i return total result = example_function(1000000) print(f"结果: {result}") ``` `timer_decorator` 是一个装饰器,它接受一个函数 `func` 作为参数,并返回一个新函数 `wrapper`。`wrapper` 函数会在调用 `func` 之前记录开始时间,调用 `func` 之后记录结束时间,最后输出函数运行时间。
2.生成器 (Generator)
生成器是一个特殊类型的迭代器,可以使用 `yield` 关键字来产生一系列值。与普通函数相比,生成器函数在每次调用 `yield` 时会保存当前的执行状态,下一次调用时会从之前保存的状态继续执行。这使得生成器非常适合于处理大量数据时节省内存。
示例:斐波那契数列生成器
下面是一个使用生成器实现的斐波那契数列:
```python def fibonacci_generator(n): a, b = 0, 1 for _ in range(n): yield a a, b = b, a + b for num in fibonacci_generator(10): print(num, end=" ") ```
`fibonacci_generator` 是一个生成器函数,它接受一个参数 `n` ,表示需要生成多少个斐波那契数。生成器每次通过 `yield` 产生一个斐波那契数。在循环中,我们可以直接使用 `for` 循环遍历生成器产生的斐波那契数。
总结:装饰器是一种用于给函数增加新功能的方法,而生成器则是一种特殊类型的迭代器,适用于处理大量数据时节省内存。这两种概念在 Python 编程中非常常见,掌握它们对于编写高效、易读的代码非常有帮助。