python - 同一图中的多个matplotlib绘图+到pdf-Python

Question

我正在绘制一些基于熊猫数据框和系列的数据。以下是我的代码的一部分。此代码给出错误。

RuntimeError: underlying C/C++ object has been deleted


from matplotlib import pyplot as plt
from matplotlib.backends.backend_pdf import PdfPages
fig = plt.figure()

dfs = df['col2'].resample('10t', how='count')
dfs.plot()
plt.show()

reg = df.groupby('col1').size()
reg.sort()
reg[-10:].plot(kind='barh')
plt.show()

pp = PdfPages('foo.pdf')
fig.savefig(pp, format='pdf') 
pp.close()

我有两个问题。

1. 如何在一个输出中绘制多个图？（在这里我为每个图获得多个输出）
2. 如何将所有这些图写入一个pdf？

我发现这是一个相关的问题。

Answer 1

以下是给我预期结果的代码部分，可能有更优雅的方法来做到这一点；

def plotGraph(X):
    fig = plt.figure()
    X.plot()
    return fig


plot1 = plotGraph(dfs)
plot2 = plotGraph2(reg[:-10])
pp = PdfPages('foo.pdf')
pp.savefig(plot1)
pp.savefig(plot2)
pp.close()

Answer 2

请参阅以下内容以使用 Pandas 定位不同的子图。

我假设您需要 2 个子图（以行方式）。因此，您的代码可以修改如下：

from matplotlib import pyplot as plt

fig, axes = plt.subplots(nrows=2)

dfs = df['col2'].resample('10t', how='count')
dfs.plot(ax=axes[0])

reg = df.groupby('col1').size()
reg.sort()
reg[-10:].plot(kind='barh',ax=axes[0])

plt.savefig('foo.pdf')

Answer 3

matplotlib默认情况下将图合并为一个图形。请参阅以下片段 -

>>> import pylab as plt
>>> randomList = [randint(0, 40) for _ in range(10)]
>>> randomListTwo = [randint(0, 40) for _ in range(10)]
>>> testFigure = plt.figure(1)
>>> plt.plot(randomList)
[<matplotlib.lines.Line2D object at 0x03F24A90>]
>>> plt.plot(randomListTwo)
[<matplotlib.lines.Line2D object at 0x030B9FB0>]
>>> plt.show()

给你一个如下图 -

此外，可以通过您发布的命令轻松地将文件保存为 PDF -

>>> from matplotlib.backends.backend_pdf import PdfPages
>>> pp = PdfPages('foo.pdf')
>>> testFigure.savefig(pp, format='pdf')
>>> pp.close()

这给了我一个类似数字的PDF。