python - 将多个绘图保存在一个 PDF 文件中

Question

绘图模块

def plotGraph(X,Y):
    fignum = random.randint(0,sys.maxint)
    plt.figure(fignum)
    ### Plotting arrangements ###
    return fignum

主模块

import matplotlib.pyplot as plt
### tempDLStats, tempDLlabels are the argument
plot1 = plotGraph(tempDLstats, tempDLlabels)
plot2 = plotGraph(tempDLstats_1, tempDLlabels_1)
plot3 = plotGraph(tempDLstats_2, tempDLlabels_2)
plt.show()

我想将所有图形 plot1、plot2、plot3 保存到一个 PDF 文件中。有没有办法实现它？我不能在主模块中包含 plotGraph 函数。

有一个名为 pyplot.savefig 的函数，但它似乎只适用于单个图形。有没有其他方法可以实现？

Answer 1

如果有人从 google 来到这里，希望将单个数字转换为 .pdf(这正是我想要的):

import matplotlib.pyplot as plt

f = plt.figure()
plt.plot(range(10), range(10), "o")
plt.show()

f.savefig("foo.pdf", bbox_inches='tight')