python - 如何存储 %%timeit 单元魔法的结果？

Question

我不知道如何存储细胞魔法的结果 - %%timeit？我读了：

1. 你能捕捉到 ipython 魔法方法的输出吗？
2. 捕获 IPython 魔术函数的结果

在这个问题中，只回答关于线魔法的问题。在行模式 ( %) 中，这有效：

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'

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