如何在Databricks笔记本中捕获单元格的输出

Question

我尝试使用：%%capture my_cap 但它给出了一个错误：

TypeError: run_cell() missing 1 required positional argument: 'raw_cell'

所以它不可用:-(还有其他选择吗？

Answer 1

在第一个单元格中包括以下内容：

from IPython.utils.capture import CapturedIO   
capture = CapturedIO(sys.stdout, sys.stderr)
...
...
# at the end of desired output:
cmem = capture.stdout

在第二个单元格中，您可以访问变量cmem：

print(cmem)