Seabornプロットをファイルに保存する方法

私は次のコード（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()

しかし、私はこのエラーを受け取ります：

  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'

私は決勝戦output.pngが存在し、次のようになることを期待しています：

どうすれば問題を解決できますか？

を削除してget_figure使用するsns_plot.savefig('output.png')

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

提案されたソリューションはSeaborn 0.8.1と互換性がありません

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

次の呼び出しにより、図にアクセスできます（Seaborn 0.8.1互換）。

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

以前にこの回答で見たように。

更新： 私は最近seabornのPairGridオブジェクトを使用して、この例のようなプロットを生成しました。この場合、GridPlotは、たとえばsns.swarmplotのようなプロットオブジェクトではないため、get_figure（）関数はありません。matplotlibの図に直接アクセスすることは可能です

fig = myGridPlotObject.fig

このスレッドの他の投稿で以前に提案されたように。

上記の解決策のいくつかは私にとってはうまくいきませんでした。それ.figを試したときに属性が見つからず、.savefig()直接使用できませんでした。しかし、うまくいったのは：

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

私は新しいPythonユーザーなので、これが更新によるものかどうかわかりません。他の誰かが私と同じ問題に遭遇した場合に備えて、それについて言及したかったのです。

直接のsavefig方法が使えればいいのですがsns_plot。

sns_plot.savefig("output.png")

コードを明確にするために、sns_plot常駐しているmatplotlib Figureにアクセスしたい場合は、次のコマンドで直接取得できます。

fig = sns_plot.fig

この場合get_figure、コードが想定している方法はありません。

とを使用distplotしget_figureて、画像を正常に保存します。

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

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

更新メモ：sizeに変更されましたheight。

これは私のために働く

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

matplotlib figureオブジェクトを作成して使用することもできます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")seaborn 0.8.1 で使用するとエラーが発生します。

代わりに：

import seaborn as sns

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

参考までに、以下のコマンドはseaborn 0.8.1で機能したので、最初の回答はまだ有効です。

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