Comment enregistrer un tracé Seaborn dans un fichier

J'ai essayé le code suivant ( 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()

Mais j'obtiens cette erreur:

  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'

J'espère que la finale output.pngexistera et ressemblera à ceci:

Comment puis-je résoudre le problème?

Retirez le get_figureet utilisez simplementsns_plot.savefig('output.png')

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

Les solutions proposées sont incompatibles avec Seaborn 0.8.1

donnant les erreurs suivantes car l'interface Seaborn a changé:

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

Les appels suivants vous permettent d'accéder à la figure (compatible Seaborn 0.8.1):

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

comme vu précédemment dans cette réponse .

MISE À JOUR: J'ai récemment utilisé l'objet PairGrid de seaborn pour générer un tracé similaire à celui de cet exemple . Dans ce cas, puisque GridPlot n'est pas un objet de tracé comme, par exemple, sns.swarmplot, il n'a pas de fonction get_figure (). Il est possible d'accéder directement à la figure matplotlib en

fig = myGridPlotObject.fig

Comme suggéré précédemment dans d'autres articles de ce fil.

Certaines des solutions ci-dessus n'ont pas fonctionné pour moi. L' .figattribut n'a pas été trouvé lorsque j'ai essayé cela et je n'ai pas pu l'utiliser .savefig()directement. Cependant, ce qui a fonctionné était:

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

Je suis un nouvel utilisateur de Python, donc je ne sais pas si cela est dû à une mise à jour. Je voulais le mentionner au cas où quelqu'un d'autre rencontrerait les mêmes problèmes que moi.

Vous devriez simplement pouvoir utiliser la savefigméthode de sns_plotdirectement.

sns_plot.savefig("output.png")

Pour plus de clarté avec votre code, si vous souhaitez accéder à la figure matplotlib qui sns_plotréside, vous pouvez l'obtenir directement avec

fig = sns_plot.fig

Dans ce cas, il n'y a pas de get_figureméthode comme le suppose votre code.

J'utilise distplotet get_figurepour enregistrer l'image avec succès.

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

Moins de lignes pour les chercheurs 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')

NOTE DE MISE À JOUR: a sizeété changé en height.

Cela fonctionne pour moi

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

Il est également possible de créer simplement un figureobjet matplotlib puis d'utiliser 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

Vous obtiendrez une erreur pour l'utilisation sns.figure.savefig("output.png")dans seaborn 0.8.1.

Utilisez plutôt:

import seaborn as sns

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

Juste pour info, la commande ci-dessous fonctionnait dans seaborn 0.8.1 donc je suppose que la réponse initiale est toujours valable.

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