在matplotlib中将多个图形保存到一个pdf文件

Question

我有一个代码可以根据 groupby 创建大约 50 个图表。代码如下所示：

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)

在matplotlib中将多个图形保存到一个pdf文件

Saving multiple figures to one pdf file in matplotlib

python

matplotlib