matplotlib画多个图并保存为PDF

有时候需要用matplotlib画很多图，一个一个保存会产生很多文件。因此可以把多个图保存到一个PDF文件里面

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages
plt.style.use('science')
plt.rcParams['figure.figsize'] = 6, 2
pdf = PdfPages('save path.pdf')
for i in range(3):
    x = np.random.random((10,))
    fig = plt.figure()
    plt.plot(x, '*-')
    plt.title(f'random {i}')
    pdf.savefig(fig)
    plt.close()
pdf.close()

上面代码中使用了

plt.style.use('science')

这是一个科学绘图库，可使画出的图更好看。详细使用可点击链接：SiciencePlots

以上代码画出的图如下图所示