代码中的%timeit等效项

马克·范德·威尔克

magic命令%timeit非常适合以交互方式测量代码执行时间。但是,我想要得到结果%timeit以便绘制结果。timeit.timeit也允许这样做,但是不具有迭代次数的自动缩放和具有的结果的标准化%timeit

是否有一个内置函数可以计时一段代码,该代码还可以自动调整其执行的迭代次数并返回归一化的结果?

塞弗特

magic%timeit命令提供了一个-o选项:

-o:返回TimeitResult,可以将其存储在变量中以更详细地检查结果。

它仍然会打印结果,但也会返回结果,以便可以将其捕获到变量中。魔术命令的语法受到一些限制,但是您可以list通过将变量分配给变量并将该变量附加到列表中来收集不同的结果

res = []
for i in range(3):
    a = %timeit -o 10*10
    res.append(a)
# 10000000 loops, best of 3: 61 ns per loop
# 10000000 loops, best of 3: 61.1 ns per loop
# 10000000 loops, best of 3: 60.8 ns per loop

然后访问res

print(res)
# [<TimeitResult : 10000000 loops, best of 3: 61.2 ns per loop>,
#  <TimeitResult : 10000000 loops, best of 3: 61.3 ns per loop>,
#  <TimeitResult : 10000000 loops, best of 3: 61.5 ns per loop>]

这些结果中的每一个都有几个属性,可能值得关注:

print(res[0].all_runs)
# [0.6166532894762563, 0.6102780388983005, 0.6370787790842183]
print(res[0].best)
# 6.102780388983005e-08
print(res[0].compile_time)
# 0.00020554513866197934
print(res[0].loops)
# 10000000
print(res[0].repeat)
# 3
print(res[0].worst)
# 1.1170931449020795e-06

例如,要绘制最佳时间,您需要创建一个包含最佳值的新列表:

res_best_times = [result.best * 1e9 for result in res] 
# "* 1e9" to get the result in nanoseconds
print(res_best_times)
# [61.2, 61.3, 61.5]

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章