如何存储 %%timeit 单元格魔术的结果？

Question

我不知道如何存储单元魔法的结果 - %%timeit？我读过：

Can you capture the output of ipython's magic methods?
Capture the result of an IPython magic function

并且在这个问题中只回答了线魔术。在行模式下 (%) 这有效：

In[1]: res = %timeit -o np.linalg.inv(A)

但在单元格模式下 (%%) 它不会:

In[2]: res = %%timeit -o 
       A = np.mat('1 2 3; 7 4 9; 5 6 1')
       np.linalg.inv(A)

它只是执行单元格，没有魔法。是错误还是我做错了什么？

Answer 1

您可以在 %%timeit -o 单元格之后使用 _ 变量（存储最后的结果）并将其分配给某个可重复使用的变量：

In[2]: %%timeit -o 
       A = np.mat('1 2 3; 7 4 9; 5 6 1')
       np.linalg.inv(A)
Out[2]: blabla
        <TimeitResult : 1 loop, best of 3: 588 µs per loop>

In[3]: res = _

In[4]: res
Out[4]: <TimeitResult : 1 loop, best of 3: 588 µs per loop>

我不认为这是一个错误，因为单元格模式命令必须是该单元格中的第一个命令，因此您不能在该命令前面放置任何内容（甚至 res = ...）。

但是您仍然需要 -o，否则 _ 变量包含 None.

Answer 2

如果你只关心单元魔法的输出，例如出于记录目的 - 并且您不需要 TimeitResult 对象中包含的额外元数据，您也可以将它与 %%capture:

结合使用

%%capture result
%%timeit

A = np.mat('1 2 3; 7 4 9; 5 6 1')
np.linalg.inv(A)

然后您可以从 result.stdout 获取输出，这将产生单元格的任何输出 - 包括计时结果。

print(result.stdout)

'26.4 us +- 329 ns per loop (mean +- std. dev. of 7 runs, 10000 loops each)\n'

这适用于任意单元格魔术，如果下划线解决方案不起作用，可以作为后备。

如何存储 %%timeit 单元格魔术的结果？

How to store the result from %%timeit cell magic?

python

ipython

ipython-magic