How to move the legend in Seaborn FacetGrid outside of the plot?

空扰寡人 提交于 2020-08-08 03:39:09

问题


I have the following code:

g = sns.FacetGrid(df, row="Type", hue="Name", size=3, aspect=3)
g = g.map(sns.plt.plot, "Volume", "Index")
g.add_legend()
sns.plt.show()

This results in the following plot:

How can I move the legend outside of the plot?


回答1:


You can do this by resizing the plots:

g = sns.FacetGrid(df, row="Type", hue="Name", size=3, aspect=3)
g = g.map(sns.plt.plot, "Volume", "Index")
for ax in g.axes.flat:
    box = ax.get_position()
    ax.set_position([box.x0,box.y0,box.width*0.9,box.height])

sns.plt.legend(loc='center left',bbox_to_anchor=(1,0.5))
sns.plt.show()

Example:

import seaborn as sns

tips = sns.load_dataset('tips')

# more informative values
condition = tips['smoker'] == 'Yes'
tips['smoking_status'] = ''
tips.loc[condition,'smoking_status'] = 'Smoker'
tips.loc[~condition,'smoking_status'] = 'Non-Smoker'

g = sns.FacetGrid(tips,row='sex',hue='smoking_status',size=3,aspect=3)
g = g.map(plt.scatter,'total_bill','tip')
for ax in g.axes.flat:
    box = ax.get_position()
    ax.set_position([box.x0,box.y0,box.width*0.85,box.height])

sns.plt.legend(loc='upper left',bbox_to_anchor=(1,0.5))
sns.plt.show()

Results in:




回答2:


Following Seaborn documentation you can add the arg legend_out=True to your call and that should fix the problem

https://seaborn.pydata.org/generated/seaborn.FacetGrid.html

Your code would then look like

g = sns.FacetGrid(df, row="Type", hue="Name", size=3, aspect=3, legend_out=True)
g = (g.map(plt.plot, "Volume", "Index").add_legend())
plt.show()



回答3:


According to mwaskom's comment above, this is a bug in OS X. Indeed switching to another backend solves the issue.

For instance, I put this into my matplotlibrc:

backend : TkAgg   # use Tk with antigrain (agg) rendering


来源:https://stackoverflow.com/questions/38773560/how-to-move-the-legend-in-seaborn-facetgrid-outside-of-the-plot

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!