Speichern von Plots (AxesSubPlot), die aus Python-Pandas mit der savefig von matplotlib generiert wurden

Ich verwende Pandas, um aus einem Datenrahmen einen Plot zu generieren, den ich in einer Datei speichern möchte:

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

Es scheint, als sollte die letzte Zeile mit matplotlibs savefig den Trick machen. Dieser Code erzeugt jedoch den folgenden Fehler:

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

Alternativ kann der Versuch, savefig direkt im Plot aufzurufen, auch folgende Fehler verursachen:

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'

Ich denke, ich muss irgendwie die von plot () zurückgegebene Nebenhandlung zu einer Figur hinzufügen, um savefig zu verwenden. Ich frage mich auch, ob dies vielleicht mit der Magie hinter der AxesSubPlot-Klasse zu tun hat .

BEARBEITEN:

Das Folgende funktioniert (ohne Fehler), hinterlässt aber ein leeres Seitenbild ....

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

EDIT 2: Der folgende Code funktioniert auch gut

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

Die gcf-Methode ist in V 0.14 entzogen. Der folgende Code funktioniert für mich:

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

Ich bin mir also nicht ganz sicher, warum dies funktioniert, aber es speichert ein Bild mit meiner Handlung:

dtf = pd.DataFrame.from_records(d,columns=h)
dtf2.plot()
fig = plt.gcf()
fig.savefig('output.png')

Ich vermute, dass der letzte Ausschnitt aus meinem ursprünglichen Beitrag leer gespeichert wurde, weil die Figur nie die von Pandas erzeugten Achsen bekommen hat. Mit dem obigen Code wird das Figurenobjekt durch den Aufruf gcf () (aktuelle Figur abrufen) aus einem magischen globalen Zustand zurückgegeben, der automatisch in Achsen backt, die in der obigen Zeile dargestellt sind.

Es scheint mir einfach, plt.savefig()Funktion für plot()Funktion zu verwenden:

import matplotlib.pyplot as plt
dtf = pd.DataFrame.from_records(d,columns=h)
dtf.plot()
plt.savefig('~/Documents/output.png')

Sie können ax.figure.savefig()wie in einem Kommentar zur Frage vorgeschlagen Folgendes verwenden:

import pandas as pd

df = pd.DataFrame([0, 1])
ax = df.plot.line()
ax.figure.savefig('demo-file.pdf')

Dies hat keinen praktischen Vorteil gegenüber ax.get_figure().savefig()den in anderen Antworten vorgeschlagenen, sodass Sie die Option auswählen können, die Sie am ästhetischsten finden. In der Tat get_figure()einfach zurückself.figure :

# Source from snippet linked above
def get_figure(self):
    """Return the `.Figure` instance the artist belongs to."""
    return self.figure

Dies kann ein einfacherer Ansatz sein:

(DesiredFigure) .get_figure (). Savefig ('figure_name.png')

dh

dfcorr.hist(bins=50).get_figure().savefig('correlation_histogram.png')