Python 计算函数运行时间

在Python中，有时候为了优化代码，我们需要监控哪些函数使用了比较多的时间，或者是为了单纯查看函数的运行时间。那么本文推荐两种监控函数运行时间的方法。

---

方法一: 在程序内使用time模块记录时间，计算使用的时间

import time

def test():
    start_time = time.time()  # 记录程序开始运行时间
    s = 0
    for i in range(1000000):
        s += 1
    end_time = time.time()  # 记录程序结束运行时间
    print('Took %f second' % (end_time - start_time))
    return s

test()

# 运行结果：
Took 0.085060 second
Out[47]: 1000000

这种方法快速简单，只需要在程序中加入三行命令即可实现运行时间的计算。

---

方法二：定义装饰器，装饰需要计算时间的函数进行监控

import time
from functools import wraps

def timefn(fn):
    """计算性能的修饰器"""
    @wraps(fn)
    def measure_time(*args, **kwargs):
        t1 = time.time()
        result = fn(*args, **kwargs)
        t2 = time.time()
        print(f"@timefn: {fn.__name__} took {t2 - t1: .5f} s")
        return result
    return measure_time


# 在定义函数前进行装饰
@timefn
def test():
    s = 0
    for i in range(1000000):
        s += 1
    return s
    
test()

# 运行结果：
@timefn: test took 0.05930 s
Out[15]: 1000000

这种方法显然更具有复用性，同时也更Pythonic。适合在程序中有多个函数需要监控时间的时候使用。需要注意的是：在Python2中获取函数名称的写法为fn.func_name，而在Python3中变为了fn.__name__。该用法涉及到装饰器的使用，想要更深入地了解可以到：
http://www.runoob.com/w3cnote/python-func-decorators.html
上面的这种写法也相当于：

timefn(test())
# 运行结果：
@timefn: test took 0.05956 s
Out[16]: <function __main__.timefn.<locals>.measure_time>

但装饰器的写法不会破坏原来函数的结构。