如何保存一个Seaborn情节到一个文件

我尝试了以下代码(test_seborn .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会存在，看起来像这样:

我该如何解决这个问题?

当前回答

也可以只创建一个matplotlib图形对象，然后使用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

2018-11-14 10:54:47

其他回答

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

更新说明:大小更改为高度。

2019-12-12 12:57:29

仅供参考，以下命令在seaborn 0.8.1中工作，所以我猜最初的答案仍然有效。

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

2018-04-04 20:39:07

在seaborn 0.8.1中使用sn .figure.savefig("output.png")会得到一个错误。

而不是使用:

import seaborn as sns

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

2018-03-03 23:34:34

也可以只创建一个matplotlib图形对象，然后使用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

2018-11-14 10:54:47

我使用distplot和get_figure成功保存图片。

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

2017-12-12 03:55:49

如何保存一个Seaborn情节到一个文件

推荐文章

最新文章

标签