Python 使用 with 上下文管理器实现代码块计时

利用 with 语句进行代码计时

Python 从 2.5 版本开始引入 with 语句，并在 2.6 版本中成为默认功能（早期版本需通过 from __future__ import with_statement 导入）。该语句最初设计用于资源访问场景，确保在发生异常时也能执行必要的清理操作，例如自动关闭文件或释放线程锁。

除了资源管理，with 语句还可以用于实现高级的计时功能。传统的计时方法通常涉及调用两次 time.time()> 并计算差值，或者使用计时装饰器包裹整个函数。然而，装饰器要求代码必须封装在函数内部，而使用上下文管理器则允许对任意一段代码块进行计时，提供了更大的灵活性。

实现自定义计时上下文管理器

要实现这一功能，可以使用 contextlib 模块中的 @contextmanager 装饰器。该装饰器允许将生成器函数转换为上下文管理器。以下是实现代码块计时的具体步骤：

1. 导入 time 和 contextlib 模块。
2. 定义一个名为 timer 的生成器函数，并使用 @contextmanager 装饰。
3. 在函数内部记录开始时间 start = time.time()。
4. 使用 yield 语句将控制权交还给 with 语句块。
5. 在 yield 之后，计算并打印耗时。

from contextlib import contextmanager
import time

@contextmanager
def timer(name): 
    start = time.time()
    yield
    print(f'[{name}] done in {time.time() - start:.2f} s')

在上述代码中，.2f 用于控制输出时间的精度，可根据需要修改为其他格式。

使用示例

定义好计时器后，可以通过 with 语句包裹任意代码段来测试其执行时间。例如，计算从 1 加到一百万的耗时：

with timer('Test'): 
    i = 0
    while i < 1000000:
        i += 1

执行上述代码后，控制台将输出类似以下结果：

[Test] done in 0.11 s

这种方法不仅简洁，而且无需修改被测试代码的结构，非常适合用于性能分析和调试。

以上就是 Python 使用 with 上下文实现计时功能的详细内容，更多关于 Python 上下文管理器的资料请关注本站其它相关文章！