首页 > 代码库 > python 将figure 写入PDF

python 将figure 写入PDF

有些时候需要检查python成图的正确性,需要将figure 输出来看看,但柑橘不是很方便,于是上网查到更好的办法是将所有figure输出到PDF文件,再检查则更加方便。在matplotlib官网看到了这个方法,简单的总结如下:

需要用到的是

matplotlib.backends.backend_pdf.PdfPages
以下给一个简略程序
from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt
pdf = PdfPages(‘cut_figure.pdf‘) #先创建一个pdf文件
plt.figure
...
...
pdf.savefig() #将图片保存在pdf文件中
plt.close()
pdf.close() #这句必须有,否则程序结束pdf文件无法打开

以上是一个将单个图片保存到pdf中,若有多个图片,则可写多个这样的步骤
那如果程序中有一个循环语句,每次循环都有一个结果图呢?那只需要将pdf.close()放在整个循环外边就可以了

python 将figure 写入PDF