python - 使用 matplotlib 的 savefig 保存从 python pandas 生成的图(AxesSubPlot)

Question

我正在使用 pandas 从数据框生成图，我想将其保存到文件中:

dtf = pd.DataFrame.from_records(d,columns=h)
fig = plt.figure()
ax = dtf2.plot()
ax = fig.add_subplot(ax)
fig.savefig('~/Documents/output.png')

似乎最后一行，使用 matplotlib 的 savefig，应该可以解决问题。但是该代码会产生以下错误:

Traceback (most recent call last):
  File "./testgraph.py", line 76, in <module>
    ax = fig.add_subplot(ax)
  File "/opt/local/Library/Frameworks/Python.framework/Versions/2.7/lib/python2.7/site-packages/matplotlib/figure.py", line 890, in add_subplot
    assert(a.get_figure() is self)
AssertionError

或者，尝试直接在绘图上调用 savefig 也会出错:

dtf2.plot().savefig('~/Documents/output.png')


  File "./testgraph.py", line 79, in <module>
    dtf2.plot().savefig('~/Documents/output.png')
AttributeError: 'AxesSubplot' object has no attribute 'savefig'

我认为我需要以某种方式将 plot() 返回的子图添加到图形中才能使用 savefig。我还想知道这是否与 magic 有关。在 AxesSubPlot 类后面。

编辑:

以下作品(没有引发错误)，但给我留下了空白页图像......

fig = plt.figure()
dtf2.plot()
fig.savefig('output.png')

编辑 2:下面的代码也可以正常工作

dtf2.plot().get_figure().savefig('output.png')

Answer 1

gcf 方法在 V 0.14 中已弃用，以下代码适用于我:

plot = dtf.plot()
fig = plot.get_figure()
fig.savefig("output.png")