pythonpandasmatplotlibvisualizationfigures

How to save lmplot in pandas


For reg plot, this is working:

sns_reg_plot = sns.regplot(x="X", y="y", data=df)
sns_reg_fig = sns_reg_plot.get_figure()
sns_reg_fig.savefig(path)

But for lm plot I get an error:

sns_lm_plot = sns.lmplot(x="X", y="y", hue="hue", data=df)
..
    sns_lm_fig = sns_lm_plot.get_figure()
AttributeError: 'FacetGrid' object has no attribute 'get_figure'

Solution

  • Just remove the .get_figure line

        sns_im_plot = sns.lmplot(x="X", y="y",hue="hue" data=df)
        sns_im_plot.savefig(path)