如何使用Matplotlib导出多图层PDF？多曲线图表分层需求

需求说明 #

需要绘制多条同步测量曲线在同一张图表中，生成支持图层切换的PDF文件，要求：

• 图层1：包含图表的非曲线元素（文本、坐标轴、网格、图例等）
• 图层2至图层n：分别对应每组测量曲线，支持阅读器单独开启/关闭

解决方案 #

Matplotlib的PDF后端支持可选内容组（OCG，即PDF图层）功能，可以实现这个需求。核心思路是为不同类型的元素指定对应的可选内容组，保存PDF时这些组会成为可独立切换的图层。

修改后的实现代码 #

import matplotlib.pyplot as plt
import random
from matplotlib.backends.backend_pdf import PdfPages

# 生成5组随机测量数据
data = []
for i in range(5):
    x = list(range(10))
    y = [random.random() for _ in range(10)]
    data.append((x, y))

# 创建PDF并定义图层
with PdfPages("layered_graph.pdf") as pdf:
    fig, ax = plt.subplots()
    
    # 获取PDF文档对象，创建可选内容组（图层）
    doc = fig.canvas.get_renderer()._doc
    # 创建非曲线元素专属图层
    non_curve_layer = doc.add_optional_content_group("非曲线元素")
    # 为每条曲线创建单独图层
    curve_layers = [doc.add_optional_content_group(f"第{i+1}组曲线") for i in range(len(data))]
    
    # 绘制非曲线元素并绑定到对应图层
    ax.set_xlabel("X轴")
    ax.set_ylabel("Y轴")
    ax.set_title("多图层同步测量曲线")
    ax.grid(True)
    # 将坐标轴、标题、网格等非曲线元素关联到指定图层
    for artist in ax.get_children():
        if not isinstance(artist, plt.Line2D):
            artist.set_optional_content_group(non_curve_layer)
    
    # 绘制每条曲线并绑定到单独图层
    for idx, (x, y) in enumerate(data):
        line, = ax.plot(x, y, label=f"曲线{idx+1}")
        line.set_optional_content_group(curve_layers[idx])
    
    # 将图例绑定到非曲线图层
    legend = ax.legend()
    legend.set_optional_content_group(non_curve_layer)
    
    # 保存带图层的PDF
    pdf.savefig(fig)
plt.close()

关键逻辑解释 #

1. 图层创建：通过doc.add_optional_content_group()创建命名图层，明确区分非曲线元素和各条曲线的分组。
2. 元素绑定：
   • 遍历坐标轴子元素，筛选出非曲线类型的元素（如坐标轴、网格、标题），绑定到"非曲线元素"图层。
   • 每条曲线绘制完成后，单独绑定到对应的曲线图层。
3. 兼容性：生成的PDF可在支持OCG的阅读器（如Adobe Acrobat、Foxit Reader）中，通过图层面板单独开关任意曲线或非曲线元素。

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