如何在 Matplotlib 中将图另存为 PDF 文件

Python 3月前 141

从 Matplotlib 生成的图可以通过使用 savefig()方法中文件名的.pdf 扩展名简单地保存为 PDF 文件。要将多个图保存在一个 PDF 文件中，我们使用 PdfPages 类。

`savefig()` 方法将图另存为 PDF 文件

我们可以使用 savefig() 方法将绘图简单地保存为 Matplotlib 中的图像文件。

savefig() 方法的语法：

matplotlib.pyplot.savefig(fname,
        dpi=None,
        facecolor='w',
        edgecolor='w',
        orientation='portrait',
        papertype=None,
        format=None,
        transparent=False,
        bbox_inches=None,
        pad_inches=0.1,
        frameon=None,
        metadata=None)

参数部分中的 fname 代表文件相对于工作目录的名称和路径。如果我们使用.pdf 作为文件名的扩展名，则该图将另存为 PDF 文件。

import numpy as np
import matplotlib.pyplot as plt
x=np.linspace(0,5,50)
y=np.cos(2 * np.pi * x)
plt.scatter(x, y)
plt.plot(x, y)
plt.title("Plot of cosx")
plt.xlabel("x")
plt.ylabel("cosx")
plt.show()
plt.savefig("Save Plot as PDF file using savefig.pdf")

这会将生成的图以文件名 Save Plot as PDF file using savefig.pdf 保存在当前工作目录中。

`PdfPages` 类的 `savefig()` 方法

如果我们希望在一个图中绘制多个图，则可以使用 PdfPages 类的 savefig() 方法。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages
x=np.linspace(-3,3,100)
y1=np.sin(x)
y2=np.cos(x)
y3=1/(1+np.exp(-x))
y4=np.exp(x)
def retFig(x, y):
    fig = plt.figure()
    a= plt.plot(x, y)
    return fig
fig1 = retFig(x, y1)
fig2 = retFig(x, y2)
fig3 = retFig(x, y3)
fig4 = retFig(x, y4)
pp = PdfPages('Save multiple plots as PDF.pdf')
pp.savefig(fig1)
pp.savefig(fig2)
pp.savefig(fig3)
pp.savefig(fig4)
pp.close()