如何保存Jupyter中display()函数展示的内容？

Jupyter的display()函数可展示各类对象（如DataFrame），其内部会生成位图、SVG或HTML资源。是否存在通用方法，将display()展示的内容保存到文件中？

目前仅见通过第三方API实现类似功能的示例（例如针对DataFrame，使用Pandas API及第三方API保存PNG或SVG，但渲染效果与Jupyter展示的不一致）：

for x, name in things:
  display(makey(x), saveAs=f"{name}.svg")

或

for x, name in things:
  display(makey(x))
  saveAsJupyterShowsIt(makey(x), saveAs=f"{name}.svg")

这种所见即所得（WYSIWYG）的方式比调用底层API更便捷、效果更一致。

以下是几种实现"所见即所得"保存display()输出的通用方法：

1. 直接调用对象的MIME渲染接口 #

Jupyter的display()依赖IPython的MIME类型渲染系统，每个可展示对象都实现了_repr_mimebundle_方法，我们可以直接获取该方法返回的渲染内容并保存：

from IPython.display import display

def save_display(obj, filename):
    # 获取对象所有可用的渲染格式
    mime_bundle, _ = obj._repr_mimebundle_(include=None, exclude=None)
    ext = filename.split('.')[-1].lower()
    
    # 映射文件后缀到对应的MIME类型
    mime_map = {
        'svg': 'image/svg+xml',
        'png': 'image/png',
        'html': 'text/html',
        'pdf': 'application/pdf',
        'json': 'application/json'
    }
    
    target_mime = mime_map.get(ext)
    if not target_mime or target_mime not in mime_bundle:
        raise ValueError(f"对象不支持导出为{ext}格式，可用格式：{[k for k, v in mime_map.items() if v in mime_bundle]}")
    
    content = mime_bundle[target_mime]
    # 处理二进制内容（如PNG、PDF）
    if isinstance(content, bytes):
        with open(filename, 'wb') as f:
            f.write(content)
    else:
        with open(filename, 'w', encoding='utf-8') as f:
            f.write(content)
    
    # 同时在Jupyter中展示内容
    display(obj)

# 使用示例：保存DataFrame为HTML
import pandas as pd
df = pd.DataFrame({'col1': [1, 2], 'col2': [3, 4]})
save_display(df, 'dataframe_output.html')

# 使用示例：保存Matplotlib图为SVG
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot([1, 3, 2])
save_display(fig, 'plot_output.svg')

2. 批量导出Notebook中的所有输出 #

如果需要保存整个Notebook里的display()内容，可以用nbconvert工具将Notebook转为HTML或PDF，完全保留Jupyter的渲染效果：

# 导出为HTML
jupyter nbconvert --to html your_notebook.ipynb

# 导出为PDF
jupyter nbconvert --to pdf your_notebook.ipynb

导出后可以直接从HTML/PDF文件中提取需要的内容，适合批量处理场景。

3. 自定义装饰器简化调用 #

可以写一个装饰器，让函数返回的对象在展示的同时自动保存：

def save_on_display(filename):
    def wrapper(obj):
        save_display(obj, filename)
        return obj
    return wrapper

# 使用示例
@save_on_display('my_figure.svg')
def generate_figure():
    import matplotlib.pyplot as plt
    fig, ax = plt.subplots()
    ax.scatter([1,2,3], [4,5,6])
    return fig

# 调用函数时会自动展示并保存
generate_figure()

内容的提问来源于stack exchange，提问作者user48956