Jak zapisać wykres Seaborn do pliku

Wypróbowałem następujący kod ( 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()

Ale pojawia się ten błąd:

  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'

Spodziewam się, że finał output.pngbędzie istniał i będzie wyglądał tak:

Jak mogę rozwiązać problem?

Usuń get_figurei po prostu użyjsns_plot.savefig('output.png')

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

Sugerowane rozwiązania są niekompatybilne z Seaborn 0.8.1

podając następujące błędy, ponieważ zmienił się interfejs Seaborn:

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

Poniższe wywołania umożliwiają dostęp do rysunku (kompatybilny z Seaborn 0.8.1):

swarm_plot = sns.swarmplot(...)
fig = swarm_plot.get_figure()
fig.savefig(...) 

jak widać wcześniej w tej odpowiedzi .

UPDATE: Niedawno użyłem obiektu PairGrid z seaborn do wygenerowania wykresu podobnego do tego w tym przykładzie . W tym przypadku, ponieważ GridPlot nie jest obiektem wykresu, jak na przykład sns.swarmplot, nie ma funkcji get_figure (). Istnieje możliwość bezpośredniego dostępu do rysunku matplotlib przez

fig = myGridPlotObject.fig

Jak sugerowano wcześniej w innych postach w tym wątku.

Niektóre z powyższych rozwiązań u mnie nie działały. .figAtrybut nie został znaleziony, gdy próbowałem tego i nie byłem w stanie używać .savefig()bezpośrednio. Jednak zadziałało:

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

Jestem nowszym użytkownikiem Pythona, więc nie wiem, czy jest to spowodowane aktualizacją. Chciałem o tym wspomnieć na wypadek, gdyby ktoś inny napotkał te same problemy co ja.

Powinieneś po prostu móc użyć savefigmetody sns_plotbezpośrednio.

sns_plot.savefig("output.png")

Dla jasności kodu, jeśli chcesz uzyskać dostęp do rysunku matplotlib, który się sns_plotznajduje, możesz go pobrać bezpośrednio za pomocą

fig = sns_plot.fig

W takim przypadku nie ma get_figuremetody, jak zakłada Twój kod.

Używam distploti, get_figureaby pomyślnie zapisać obrazek.

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

Mniej wierszy dla wyszukiwarek 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')

UPDATE NOTE: sizezostała zmieniona na height.

To działa dla mnie

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')

Można również po prostu utworzyć figureobiekt matplotlib, a następnie użyć plt.savefig(...):

from matplotlib import pyplot as plt
import seaborn as sns
import pandas as pd

df = sns.load_dataset('iris')
plt.figure() # Push new figure on stack
sns_plot = sns.pairplot(df, hue='species', size=2.5)
plt.savefig('output.png') # Save that figure

sns.figure.savefig("output.png")Wystąpiłby błąd podczas używania w wersji 0.8.1 Seaaborn.

Zamiast tego użyj:

import seaborn as sns

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

Do Twojej wiadomości, poniższe polecenie działało w wersji 0.8.1 na seaborn, więc myślę, że początkowa odpowiedź jest nadal aktualna.

sns_plot = sns.pairplot(data, hue='species', size=3)
sns_plot.savefig("output.png")