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

Question

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

1. Can you capture the output of ipython's magic methods?
2. 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。