Come salvare un diagramma di Seaborn in un file

Ho provato il seguente codice ( 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()

Ma ottengo questo errore:

  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'

Mi aspetto che il finale output.pngesista e assomigli a questo:

Come posso risolvere il problema?

Rimuovere il get_figuree basta usaresns_plot.savefig('output.png')

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

Le soluzioni suggerite sono incompatibili con Seaborn 0.8.1

dando i seguenti errori perché l'interfaccia Seaborn è cambiata:

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

Le seguenti chiamate consentono di accedere alla figura (compatibile con Seaborn 0.8.1):

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

come visto in precedenza in questa risposta .

AGGIORNAMENTO: Di recente ho usato l'oggetto PairGrid di Seaborn per generare un diagramma simile a quello in questo esempio . In questo caso, poiché GridPlot non è un oggetto trama come, ad esempio, sns.swarmplot, non ha alcuna funzione get_figure (). È possibile accedere direttamente alla figura matplotlib tramite

fig = myGridPlotObject.fig

Come precedentemente suggerito in altri post di questa discussione.

Alcune delle soluzioni di cui sopra non hanno funzionato per me. L' .figattributo non è stato trovato quando l'ho provato e non sono stato in grado di utilizzare .savefig()direttamente. Tuttavia, ciò che ha funzionato è stato:

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

Sono un nuovo utente di Python, quindi non so se ciò sia dovuto a un aggiornamento. Volevo menzionarlo nel caso in cui qualcun altro incontri gli stessi problemi che ho fatto io.

Dovresti essere in grado di usare il savefigsns_plot direttamente metodo di .

sns_plot.savefig("output.png")

Per chiarezza con il codice se si desidera accedere alla figura matplotlib che sns_plotrisiede in, è possibile ottenerlo direttamente con

fig = sns_plot.fig

In questo caso non esiste un get_figuremetodo come presuppone il codice.

Io uso distplote get_figureper salvare l'immagine con successo.

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

Meno linee per i ricercatori del 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')

NOTA AGGIORNAMENTO: è sizestato modificato in height.

Questo funziona per me

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

È anche possibile creare un figureoggetto matplotlib e quindi utilizzare 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

Si otterrebbe un errore per l'utilizzo sns.figure.savefig("output.png")in Seaborn 0.8.1.

Invece usa:

import seaborn as sns

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

Cordiali saluti, il comando seguente ha funzionato in Seaborn 0.8.1, quindi immagino che la risposta iniziale sia ancora valida.

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