Cómo guardar una trama de Seaborn en un archivo

Intenté el siguiente código ( 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()

Pero me sale este error:

  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'

Espero que la final output.pngexista y se vea así:

¿Cómo puedo resolver el problema?

Retire el get_figurey simplemente usesns_plot.savefig('output.png')

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

Las soluciones sugeridas son incompatibles con Seaborn 0.8.1

dando los siguientes errores porque la interfaz Seaborn ha cambiado:

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

Las siguientes llamadas le permiten acceder a la figura (compatible con Seaborn 0.8.1):

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

como se vio anteriormente en esta respuesta .

ACTUALIZACIÓN: Recientemente he usado el objeto PairGrid de seaborn para generar una trama similar a la de este ejemplo . En este caso, dado que GridPlot no es un objeto de trazado como, por ejemplo, sns.swarmplot, no tiene la función get_figure (). Es posible acceder directamente a la figura matplotlib mediante

fig = myGridPlotObject.fig

Como se sugirió anteriormente en otras publicaciones de este hilo.

Algunas de las soluciones anteriores no funcionaron para mí. El .figatributo no se encontró cuando lo intenté y no pude usarlo .savefig()directamente. Sin embargo, lo que funcionó fue:

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

Soy un nuevo usuario de Python, por lo que no sé si esto se debe a una actualización. Quería mencionarlo en caso de que alguien más se encuentre con los mismos problemas que yo.

Debería poder utilizar el savefigmétodo sns_plotdirectamente.

sns_plot.savefig("output.png")

Para mayor claridad con su código, si desea acceder a la figura matplotlib que sns_plotreside, puede obtenerlo directamente con

fig = sns_plot.fig

En este caso, no hay ningún get_figuremétodo como se supone en su código.

Yo uso distploty get_figurepara guardar la imagen con éxito.

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

Menos líneas para los buscadores de 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 DE ACTUALIZACIÓN: sizese cambió a height.

Esto funciona para mi

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

También es posible crear un figureobjeto matplotlib y luego usar 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")Obtendría un error al usar seaborn sea 0.8.1.

En su lugar use:

import seaborn as sns

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

Solo para su información, el siguiente comando funcionó en seaborn 0.8.1, así que supongo que la respuesta inicial sigue siendo válida.

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