如何将Seaborn绘图保存到文件中

Question

我尝试了以下代码（test\u 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'

我期望最终的输出。png将存在，并如下所示：

我如何解决这个问题？




Answer 1

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。

Answer 2

上面的一些解决方案对我不起作用。。图属性在我尝试该属性时未找到，并且无法使用。直接保存图（）。然而，所起的作用是：

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

我是一个较新的Python用户，所以我不知道这是否是由于更新。我想提一下，以防其他人遇到和我一样的问题。

Answer 3

建议的解决方案与Seaborn 0.8不兼容。1.

给出以下错误，因为Seborn接口已更改：

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的plot对象。Swarmlot，它没有get_figure（）函数。可以通过以下方式直接访问matplotlib图形：

fig = myGridPlotObject.fig

就像之前在本帖其他帖子中建议的那样。