Speichern mehrerer Zahlen in einer PDF-Datei in Matplotlib

Question

Ich habe einen Code, der etwa 50 Grafiken basierend auf groupby erstellt. Der Code sieht wie folgt aus:

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) 

Dies ist nur eine Zahl zu speichern, (die letzte in meiner Serie), wenn ich alle meine Figuren möchte in einem pdf gespeichert werden. Jede Hilfe wäre willkommen.

Answer 1

In Ihrem Code ist ein Einrückungsfehler aufgetreten. Da Ihr Plotbefehl nicht in der Schleife war, wird nur der letzte Plot erstellt.

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)