Matplotlib.pyplot: сохранить графики в pdf

------ модуль построения страниц ------

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

------ модуль построения страниц ------

----- mainModule ----

  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 в mainModule.

Есть функция с именем "pylab.savefig", но это работает, только если она размещена вместе с модулем построения. Есть ли другой способ сделать это? Предложите мне какие-либо изменения в моих функциональных кодах, чтобы я мог сохранить графики в один файл PDF.

Ответы

Ответ 1

Не обращай внимания на то, как это сделать.

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

------ модуль построения страниц ------

----- mainModule ----

 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.savefig('plot1.png')
 plot2.savefig('plot2.png')
 plot3.savefig('plot3.png')

----- mainModule -----

Ответ 2

Для нескольких графиков в одном файле PDF вы можете использовать PdfPages

В функции plotGraph вы должны вернуть фигуру и вызывать savefig объекта фигуры.

------ модуль построения страниц ------

def plotGraph(X,Y):
      fig = plt.figure()
      ### Plotting arrangements ###
      return fig

------ модуль построения страниц ------

----- mainModule ----

from matplotlib.backends.backend_pdf import PdfPages

plot1 = plotGraph(tempDLstats, tempDLlabels)
plot2 = plotGraph(tempDLstats_1, tempDLlabels_1)
plot3 = plotGraph(tempDLstats_2, tempDLlabels_2)

pp = PdfPages('foo.pdf')
pp.savefig(plot1)
pp.savefig(plot2)
pp.savefig(plot3)
pp.close()

Ответ 3

Если кто-то попадает сюда из 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')

Ответ 4

import datetime
import numpy as np
from matplotlib.backends.backend_pdf import PdfPages
import matplotlib.pyplot as plt

# Create the PdfPages object to which we will save the pages:
# The with statement makes sure that the PdfPages object is closed properly at
# the end of the block, even if an Exception occurs.
with PdfPages('multipage_pdf.pdf') as pdf:
    plt.figure(figsize=(3, 3))
    plt.plot(range(7), [3, 1, 4, 1, 5, 9, 2], 'r-o')
    plt.title('Page One')
    pdf.savefig()  # saves the current figure into a pdf page
    plt.close()

    plt.rc('text', usetex=True)
    plt.figure(figsize=(8, 6))
    x = np.arange(0, 5, 0.1)
    plt.plot(x, np.sin(x), 'b-')
    plt.title('Page Two')
    pdf.savefig()
    plt.close()

    plt.rc('text', usetex=False)
    fig = plt.figure(figsize=(4, 5))
    plt.plot(x, x*x, 'ko')
    plt.title('Page Three')
    pdf.savefig(fig)  # or you can pass a Figure object to pdf.savefig
    plt.close()

    # We can also set the file metadata via the PdfPages object:
    d = pdf.infodict()
    d['Title'] = 'Multipage PDF Example'
    d['Author'] = u'Jouni K. Sepp\xe4nen'
    d['Subject'] = 'How to create a multipage pdf file and set its metadata'
    d['Keywords'] = 'PdfPages multipage keywords author title subject'
    d['CreationDate'] = datetime.datetime(2009, 11, 13)
    d['ModDate'] = datetime.datetime.today()