Salvataggio di più cifre in un file PDF in matplotlib

Question

Ho un codice che crea circa 50 grafici basati su groupby. Il codice è simile al seguente:

import matplotlib.pyplot as plt 
from matplotlib.backends.backend_pdf import PdfPages 

with PdfPages('foo.pdf') as pdf: 
    for i, group in df.groupby('station_id'): 
     plt.figure() 


fig=group.plot(x='year', y='Value',title=str(i)).get_figure() 
pdf.savefig(fig) 

Questo sta salvando una sola figura, (l'ultimo nella mia serie) quando vorrei tutti i miei dati da memorizzare in un unico PDF. Qualsiasi aiuto sarebbe apprezzato.

Answer 1

C'è un errore di indentazione nel codice. Poiché il comando di stampa non era nel ciclo, creerà solo l'ultimo grafico.

import matplotlib.pyplot as plt 
from matplotlib.backends.backend_pdf import PdfPages 

with PdfPages('foo.pdf') as pdf: 
    for i, group in df.groupby('station_id'): 
     plt.figure() 
     fig=group.plot(x='year', y='Value',title=str(i)).get_figure() 
     pdf.savefig(fig)