Python 函数运行时间装饰器

最新推荐文章于 2024-07-17 21:51:07 发布

Inotime

最新推荐文章于 2024-07-17 21:51:07 发布

阅读量8.9k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/lnotime/article/details/81844988

Python 专栏收录该内容

43 篇文章

订阅专栏

本文介绍了一个简单的Python装饰器，用于测量并打印函数的运行时间。通过实例演示了如何使用该装饰器来评估代码片段的执行效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Python3.6.3

import time

__all__ = ['print_time']


def print_time(f):
    """Decorator of viewing function runtime.
    eg:
        ```py
        from print_time import print_time as pt


        @pt
        def work(...):
            print('work is running')


        word()

        # work is running
        # --> RUN TIME: <work> : 2.8371810913085938e-05
        ```
    """
    def fi(*args, **kwargs):
        s = time.time()
        res = f(*args, **kwargs)
        print('--> RUN TIME: <%s> : %s' % (f.__name__, time.time() - s))
        return res

    return fi


# test
@print_time
def _test1():
    print('work is running')


if __name__ == '__main__':
    _test1()

放到Lib里面设为只读方便使用