ipython - 你能捕捉到 ipython 魔法方法的输出吗？ (时间)

Question

我想捕获并绘制 5 个左右的结果 timeit调用以对数递增的 N 大小来显示 methodX()随输入缩放。

到目前为止，我已经尝试过:

output = %timeit -r 10 results = methodX(N)

这没用...

在文档中也找不到信息。我觉得您至少应该能够拦截打印的字符串。之后我可以解析它以提取我的信息。

有没有人这样做或尝试过？

PS:这是在 ipython 笔记本中，如果有差异的话。

Answer 1

这个重复的问题 Capture the result of an IPython magic function有一个 answer表明这已经实现。

调用%timeit -o 的魔法选项如:

%timeit -o <statement>

返回 TimeitResult对象，它是一个简单对象，包含有关 %timeit 的所有信息作为属性运行。例如:

In [1]: result = %timeit -o 1 + 2
Out[1]: 10000000 loops, best of 3: 23.2 ns per loop

In [2]: result.best
Out[2]: 2.3192405700683594e-08