如何知道python代码中一个函数的耗时 性能测试 cProfile初探

下面是一段利用python装饰器进行计时的代码(侵删,出处忘了):

def timerfunc(func):
    """
    A timer decorator
    """

    def function_timer(*args, **kwargs):
        """
        A nested function for timing other functions
        """
        start = time.time()
        value = func(*args, **kwargs)
        end = time.time()
        runtime = end - start
        msg = "The runtime for {func} took {time} seconds to complete"
        print(msg.format(func=func.__name__, time=runtime))
        return value

    return function_timer

利用装饰器计时是比较优雅的,但是也有下图的问题。
在这里插入图片描述
有的函数,我们明知它是核心逻辑,但因为调用次数多、每次耗时短,而不能精确测量;此外,记录数过多,不利于分析。
cProfile可以解决上一段提到的问题,它会统计调用次数和总耗时,可以将结果按总耗时排序。代码如下:

    from cProfile import Profile
    import io
	import pstats
	
    prof = Profile()
    prof.enable()
    #do something
    prof.create_stats()
    s = io.StringIO()
    ps = pstats.Stats(prof, stream=s).sort_stats(pstats.SortKey.CUMULATIVE)
    ps.print_stats()
    fp = open('profile.txt', 'w')
    fp.write(s.getvalue())
    fp.close()

说实话,cProfile我也不太满意,即使依据耗时排序,依然会有很多无关内容。我只想显示其中的用户自定义的底层函数的耗时。为此,我觉得可以将cProfile的结果输出到文件,再做后续处理。
可以先将所有想测试的函数名称放在一个list里;
然后遍历上一步中profile.txt中的每一行,检查是否是想测试的函数;
若是则输出。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值