在 matplotlib 中将多个图形保存到一个 PDF 文件

matplotlibpythondata visualization

要一次将多个图形保存到一个 PDF 文件中，我们可以采取以下步骤

步骤

设置图形大小并调整子图之间和周围的填充。
使用 figure() 方法创建一个新图形 (fig1) 或激活现有图形。
使用 plot() 方法绘制第一条线。
使用 figure() 方法创建另一个图形 (fig2) 或激活现有图形。
使用 plot() 方法绘制第二条线。
初始化一个变量， filename，制作一个pdf文件。
创建一个用户定义函数save_multi_image()，将多张图片保存在一个PDF文件中。
使用filename调用save_multi_image()函数。
创建一个新的PdfPages对象。
获取打开的图形数量。
迭代打开的图形并将其保存到文件中。
关闭创建的PDF对象。

示例

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

plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

fig1 = plt.figure()
plt.plot([2, 1, 7, 1, 2], color='red', lw=5)

fig2 = plt.figure()
plt.plot([3, 5, 1, 5, 3], color='green', lw=5)

def save_multi_image(filename):
    pp = PdfPages(filename)
    fig_nums = plt.get_fignums()
    figs = [plt.figure(n) for n in fig_nums]
    for fig in figs:
        fig.savefig(pp, format='pdf')
    pp.close()

filename = "multi.pdf"
save_multi_image(filename)