使用Python计算函数运行的耗时

258 阅读1分钟

一般方法就是在函数的开始和结束处打时间戳,然后后者减去前者就是所用耗时。更快捷的方式是使用装饰器实现。

import time

def get_time(f):

    def inner(*arg,**kwarg):
        s_time = time.time()
        res = f(*arg,**kwarg)
        e_time = time.time()
        print('耗时{}秒'.format(e_time - s_time))
        return res
    return inner

@get_time
def test():
    time.sleep(2)  

test()  # 输出:耗时2.015456199645996秒

在指定的函数上添加@get_time即可,函数执行结束后会自动输出get_time函数中的print语句。