python定义一个装饰器自动测量函数的运行时间

最新推荐文章于 2022-03-04 16:42:13 发布

原创最新推荐文章于 2022-03-04 16:42:13 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #装饰器 #运行时间

python 专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一个使用Python装饰器实现的计时功能，通过该装饰器可以测量函数的执行时间，具体展示了如何定义装饰器并应用到目标函数上，以获取函数运行所需的时间。

部署运行你感兴趣的模型镜像

#!/usr/bin/python
# -*-coding:UTF-8-*-

import time
from functools import wraps


def timefn(fn):
    @wraps(fn)
    def measure_time(*args, **kwargs):
        start_time = time.time()
        result = fn(*args, **kwargs)
        end_time = time.time()
        print ("@timefn: %s took %s seconds" % (fn.func_name, str(end_time - start_time)))
        return result

    return measure_time


@timefn
def test_time(num):
    time.sleep(1)
    return num + 5


if __name__ == '__main__':
    print(test_time(2))

您可能感兴趣的与本文相关的镜像