python 装饰器分析优化每行代码的性能

最新推荐文章于 2025-12-23 20:34:05 发布

原创最新推荐文章于 2025-12-23 20:34:05 发布 · 175 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

python 专栏收录该内容

18 篇文章

订阅专栏

该文介绍了一个用于分析Python函数运行时间的工具LineProfiler，通过装饰器`line_used_time_parse`集成到代码中，启用后可以详细显示每个代码行的执行时间，帮助优化性能。

import sys
from functools import wraps
from termcolor import cprint
from line_profiler import LineProfiler


def line_used_time_parse(func):
    @wraps(func)
    def wrap(*args, **kwargs):
        profile = LineProfiler(func)
        profile.enable()
        cprint(f"start run:{func.__name__}".center(30, "="), "green")
        res = func(*args, **kwargs)
        profile.disable()
        profile.print_stats(sys.stdout, output_unit=1e-3)
        return res
    return wrap