So speichern Sie einen Seaborn-Plot in einer Datei

Ich habe folgenden Code versucht (test_seaborn.py):

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
matplotlib.style.use('ggplot')
import seaborn as sns
sns.set()
df = sns.load_dataset('iris')
sns_plot = sns.pairplot(df, hue="species", size=2.5)
fig = sns_plot.get_figure()
fig.savefig("output.png")
#sns.plt.show()

Aber ich bekomme diesen Fehler:

  Traceback (most recent call last):
  File "test_searborn.py", line 11, in <module>
    fig = sns_plot.get_figure()
AttributeError: 'PairGrid' object has no attribute 'get_figure'

Ich erwarte das Finale output.png wird existieren und so aussehen:

Wie kann ich das Problem lösen?

Die folgenden Aufrufe ermöglichen Ihnen den Zugriff auf die Abbildung (Seaborn 0.8.1 kompatibel):

swarm_plot = sns.swarmplot(...)
fig = swarm_plot.get_figure()
fig.savefig("out.png") 

wie zuvor in dieser Antwort gesehen.

Die vorgeschlagenen Lösungen sind mit Seaborn 0.8.1 nicht kompatibel. Sie geben die folgenden Fehler aus, weil sich die Seaborn-Schnittstelle geändert hat:

AttributeError: 'AxesSubplot' object has no attribute 'fig'
When trying to access the figure

AttributeError: 'AxesSubplot' object has no attribute 'savefig'
when trying to use the savefig directly as a function

AKTUALISIEREN:
Ich habe vor kurzem verwendet PairGrid Objekt von Seaborn, um ein Diagramm zu erstellen, das dem in ähnelt dieses Beispiel. In diesem Fall seit GridPlot ist kein Handlungsobjekt wie zum Beispiel sns.swarmplotEs hat keine get_figure() Funktion. Es ist möglich, direkt auf die Matplotlib-Abbildung zuzugreifen:

fig = myGridPlotObject.fig

Einige der oben genannten Lösungen haben bei mir nicht funktioniert. Das .fig Das Attribut wurde nicht gefunden, als ich das versuchte, und ich konnte es nicht verwenden .savefig() direkt. Was jedoch funktionierte, war:

sns_plot.figure.savefig("output.png")

Ich bin ein neuerer Python-Benutzer, daher weiß ich nicht, ob dies auf ein Update zurückzuführen ist. Ich wollte es erwähnen, falls jemand auf die gleichen Probleme stößt wie ich.

Weniger Zeilen für Suchende im Jahr 2019:

import matplotlib.pyplot as plt
import seaborn as sns

df = sns.load_dataset('iris')
sns_plot = sns.pairplot(df, hue="species", height=2.5)
plt.savefig('output.png')

AKTUALISIERUNGSHINWEIS: size geändert wurde height.

Sie sollten nur in der Lage sein, die zu verwenden savefig Methode von sns_plot direkt.

sns_plot.savefig("output.png")

Zur Verdeutlichung Ihres Codes, wenn Sie auf die Matplotlib-Abbildung zugreifen möchten sns_plot wohnt dann kannst du es direkt mit bekommen

fig = sns_plot.fig

In diesem Fall gibt es keine get_figure Methode, wie Ihr Code annimmt.

ich benutze distplot und get_figure Bild erfolgreich zu speichern.

sns_hist = sns.distplot(df_train['SalePrice'])
fig = sns_hist.get_figure()
fig.savefig('hist.png')

Ich konnte die anderen Antworten nicht zum Laufen bringen und habe dies schließlich für matplotlib==3.2.1 zum Laufen gebracht. Dies gilt insbesondere, wenn Sie dies innerhalb einer for-Schleife oder eines iterativen Ansatzes tun.

sns.scatterplot(
    data=df_hourly, x="date_week", y="value",hue="variable"
)
plt.savefig('./f1.png')
plt.show()

Beachten Sie, dass die savefig vor dem Show-Call stehen muss. Andernfalls wird ein leeres Bild gespeichert.

Das funktioniert für mich

import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline

sns.factorplot(x='holiday',data=data,kind='count',size=5,aspect=1)
plt.savefig('holiday-vs-count.png')