如何在 Databricks 筆記本中捕獲單元的 output

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)