Como salvar uma plotagem Seaborn em um arquivo

Eu tentei o seguinte 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()

Mas eu recebo este erro:

  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 a final output.pngexista e fique assim:

Como posso resolver o problema?

Remova o get_figuree apenas 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")

As soluções sugeridas são incompatíveis com Seaborn 0.8.1

dando os seguintes erros porque a interface Seaborn mudou:

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

As chamadas a seguir permitem acessar a figura (compatível com Seaborn 0.8.1):

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

como visto anteriormente em nesta resposta .

UPDATE: Recentemente, usei o objeto PairGrid de seaborn para gerar um gráfico semelhante ao deste exemplo . Nesse caso, como o GridPlot não é um objeto de plotagem como, por exemplo, sns.swarmplot, ele não possui a função get_figure (). É possível acessar diretamente a figura do matplotlib

fig = myGridPlotObject.fig

Como sugerido anteriormente em outros posts deste tópico.

Algumas das soluções acima não funcionaram para mim. O .figatributo não foi encontrado quando tentei e não consegui usá-lo .savefig()diretamente. No entanto, o que funcionou foi:

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

Eu sou um usuário Python mais novo, então não sei se isso se deve a uma atualização. Eu queria mencionar isso no caso de alguém encontrar os mesmos problemas que eu.

Você deve conseguir usar o savefigmétodo sns_plotdiretamente.

sns_plot.savefig("output.png")

Para maior clareza do seu código, se você deseja acessar a figura matplotlib que sns_plotreside, é possível obtê-la diretamente com

fig = sns_plot.fig

Nesse caso, não há get_figuremétodo como o seu código assume.

Eu uso distplote get_figurepara salvar a foto com sucesso.

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

Menos linhas para os pesquisadores 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')

ATUALIZAÇÃO NOTA: sizefoi alterada para height.

Isso funciona para mim

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

Também é possível apenas criar um figureobjeto matplotlib e depois 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

Você receberia um erro ao usar sns.figure.savefig("output.png")no seaborn 0.8.1.

Em vez disso, 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")

Apenas para sua informação, o comando abaixo funcionou no seaborn 0.8.1, então acho que a resposta inicial ainda é válida.

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