我目前有一些这样的代码：
import numpy as np
import matplotlib.pyplot as plt
def generate_data():
return np.random.randint(10, size=10)
plt.figure(1)
for i in range(4):
plt.subplot(2, 2, i + 1)
plt.plot(generate_data())
plt.savefig("C:\\test.pdf")
假设如果要生成 5 个这样的图形，每个图形有 4 张图片。如何将它们保存到具有相同页面的 pdf 中。
我在这里阅读了一些关于如何保存数字的答案。比如这个
Python saving multiple figures into one PDF file
但是，我不明白如何创建所需的数字。
【问题讨论】：
标签： python matplotlib
结合您的代码和您提供的链接，这样可以节省一个 pdf (output.pdf) 的 5 页，并且在每一页上都有一个图形：
import matplotlib.backends.backend_pdf
pdf = matplotlib.backends.backend_pdf.PdfPages("output.pdf")
import numpy as np
import matplotlib.pyplot as plt
def generate_data():
return np.random.randint(10, size=10)
figs = []
n_figs = 5
for j in range(n_figs): # create all figures
plt.figure(j)
plt.suptitle("figure {}" .format(j+1))
for i in range(4):
plt.subplot(2, 2, i + 1)
plt.plot(generate_data())
for fig in range(0, plt.gcf().number + 1): # loop over all figures
pdf.savefig( fig ) # save each figure in the pdf
pdf.close()
编辑：在评论之后，这是在生成大量图形时尝试使用更少内存的版本。我们的想法是在创建每个图形后保存它，然后将其关闭。
for j in range(n_figs): # create all figures
plt.figure(j)
plt.suptitle("figure {}" .format(j+1))
for i in range(4):
plt.subplot(2, 2, i + 1)
plt.plot(generate_data())
pdf.savefig(j) # save on the fly
plt.close() # close figure once saved
pdf.close()
EDIT2：这是第三个版本，我不确定哪个更节省内存。现在的想法是，与其创建和关闭许多图形，不如只创建一个、保存、清除并重用它。
plt.figure(1) # create figure outside loop
for j in range(n_figs): # create all figures
plt.suptitle("figure {}" .format(j+1))
for i in range(4):
plt.subplot(2, 2, i + 1)
plt.plot(generate_data())
pdf.savefig(1) # save on the fly
plt.clf() # clear figure once saved
pdf.close()
【讨论】：
像这样。
fig, ax = plt.subplots(2, 2)
for i in range(5):
ax[i].plot(generate_data())
fig.savefig("C:\\test.pdf")
【讨论】：