Google Colab中自定义Altair图表另存为的文件名

Altair在Colab环境的默认渲染逻辑会将SVG导出按钮的下载文件名硬编码为visualization.svg，批量生成数十张图表时逐张重命名效率极低，通过自定义Colab渲染器即可实现渲染图表时直接指定导出文件名的需求，完全匹配预期的调用形式，且不需要改变手动点击导出的操作习惯。

实现步骤 #

1. 注册自定义渲染器 #

在笔记本开头运行以下代码，注册支持自定义文件名的渲染器，替换默认的Colab渲染逻辑：

import altair as alt

# 注册支持自定义导出文件名的Colab专用渲染器
def colab_custom_filename_renderer(spec):
    # 读取传入的自定义文件名，未指定时回退到默认文件名
    file_name = spec.pop("downloadFileName", "visualization")
    return (
        {"application/vnd.vegalite.v5+json": spec},
        {
            "application/vnd.vegalite.v5+json": {
                "embed_options": {
                    "actions": {"svg": {"filename": file_name}}
                }
            }
        },
    )

alt.renderers.register("colab_custom_dl", colab_custom_filename_renderer)
# 启用自定义渲染器
alt.renderers.enable("colab_custom_dl")

2. 按原有逻辑批量生成、展示图表 #

渲染器配置完成后，原有循环代码几乎不需要修改，在调用display()时传入downloadFileName参数即可，点击图表右上角的「导出为SVG」按钮时，会自动使用传入的文件名保存，不需要手动重命名：

# 示例调用，替换为自己的字段列表和制图逻辑即可
fields = ["销售额", "用户量", "转化率"]

def make_chart(field):
    # 此处替换为实际的制图代码
    import pandas as pd
    demo_df = pd.DataFrame({"类别": ["A", "B", "C"], "数值": [12, 25, 18]})
    return alt.Chart(demo_df).mark_bar().encode(
        x="类别", y="数值"
    ).properties(title=f"{field}统计图表")

for field in fields:
    chart = make_chart(field)
    # 传入自定义导出文件名
    chart.display(downloadFileName=field)

补充说明 #

• 该方案完全保留原有操作习惯：不需要写自动批量下载的代码，逐张核对图表后点导出按钮即可，文件名自动按配置生成
• 如果需要同时支持PNG格式导出自定义文件名，只需要在渲染器的embed_options.actions配置中增加"png": {"filename": file_name}即可
• 传入文件名时不需要手动添加.svg后缀，Vega渲染层会自动补全对应格式的后缀，不会出现重复后缀问题
• 自定义渲染器只需要在笔记本初始化时运行一次，后续所有图表展示都支持自定义文件名参数

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