Python 使用with上下文实现计时功能

引言

with 语句是从 Python 2.5 开始引入的一种与异常处理相关的功能（2.5 版本中要通过 from __future__ import with_statement 导入后才可以使用），从 2.6 版本开始缺省可用（参考 What's new in Python 2.6? 中 with 语句相关部分介绍）。with 语句适用于对资源进行访问的场合，确保不管使用过程中是否发生异常都会执行必要的“清理”操作，释放资源，比如文件使用后自动关闭、线程中锁的自动获取和释放等。

在Python常常需要记录一段代码运行了多长时间，普通的一般用两个 time.time() ，然后计算差值；高级一点的，就是写一个计时装饰器，用装饰器包裹方法，记录运行时间，但这这意味着你的代码需要写在函数里，然后才能使用此计时装饰器。

今天我要介绍的是一种可以给一段代码计时，而不是一个函数计时的方便的、“高级的”的Python计时方法。

代码如下：

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

使用方法：

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

输出：

[Test] done in 0.11 s

从1加到1百万需要0.11s。可以在 timer 通过修改 .2f 来修改时间精度。

总结