Сохранение нескольких фигур в один файл pdf в matplotlib

Question

У меня есть код, который создает около 50 графиков на основе groupby. Код выглядит следующим образом:

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) 

Это спасает только одна фигура, (последний в моей серии), когда я хотел бы все мои цифры должны быть сохранены в один PDF. Любая помощь будет оценена по достоинству.

Answer 1

В коде есть ошибка с отступом. Поскольку ваша команда построения графика не была в цикле, она создаст только последний график.

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)