如何将seaborn图例拆分为多列？

Posted 2023-02-16

【中文标题】如何将seaborn图例拆分为多列？

【英文标题】：How to split seaborn legend into multiple columns?

【发布时间】：2019-10-27 18:26:03

【问题描述】：

我使用具有不同色调和风格的 relplot，并希望显示各自的图例条目，而不是彼此下方。

所以目前我得到一个这样的传说：

相反，我希望有一个像这样的单个图例：

如何做到这一点？

我尝试设置以下但没有效果：

plot._legend
leg._ncol = 2
leg.handleheight = 1  # restricting the height

---

解决此问题的最小工作示例：

import pandas as pd
import seaborn as sns

columns = ['category1', 'category2', 'category3', 'time', 'value']

data = [['content1', 'other1', 'critera1', 0, 0.1], ['content1', 'other1', 'critera1', 1, 0.4], ['content1', 'other1', 'critera1', 2, 0.7], ['content2', 'other1', 'critera1', 0, 0.2], ['content2', 'other1', 'critera1', 1, 0.6], ['content2', 'other1', 'critera1', 2, 0.8], ['content1', 'other2', 'critera1', 0, 0.0], ['content1', 'other2', 'critera1', 1, 0.2], ['content1', 'other2', 'critera1', 2, 0.8], ['content2', 'other2', 'critera1', 0, 0.3], ['content2', 'other2', 'critera1', 1, 0.6], ['content2', 'other2', 'critera1', 2, 0.5], [
    'content1', 'other1', 'critera2', 0, 0.1], ['content1', 'other1', 'critera2', 1, 0.4], ['content1', 'other1', 'critera2', 2, 0.7], ['content2', 'other1', 'critera2', 0, 0.2], ['content2', 'other1', 'critera2', 1, 0.6], ['content2', 'other1', 'critera2', 2, 0.8], ['content1', 'other2', 'critera2', 0, 0.0], ['content1', 'other2', 'critera2', 1, 0.2], ['content1', 'other2', 'critera2', 2, 0.8], ['content2', 'other2', 'critera2', 0, 0.3], ['content2', 'other2', 'critera2', 1, 0.6], ['content2', 'other2', 'critera2', 2, 0.5], ]

df = pd.DataFrame(data, columns=columns)

plot = sns.relplot(x='time', y='value', col='category3', hue='category1', style='category2', kind="line", col_wrap=2, data=df)

leg = plot._legend
leg.set_bbox_to_anchor((0.5, 1.3, 0, 0))
leg._loc = 9

---

【问题讨论】：

没有内置的方法可以做到这一点。即使可以动态更改列数，它看起来也会与您的预期不同。因此，您需要从头开始重新创建图例。如果您需要有关如何做到这一点的答案，最好向潜在的回答者提供minimal reproducible example，他们可以用来展示解决方案。

Separate seaborn legend into two distinct boxes的可能重复

@DizietAsahi 感谢您的链接，但它们应该仍然在同一个图例中。

@ImportanceOfBeingErnest 我添加了一个 MWE。

【参考方案1】：

由于您似乎想将图例放在情节上方，我会指示 seaborn 不要使用 legend_out=False 在右侧为图例保留空间。然后只需获取seaborn创建的句柄和标签，并使用ncol=2生成新的图例即可。请注意，这只有在两列中的元素数量相同时才能正常工作，否则事情会变得一团糟。

plot = sns.relplot(x='time', y='value', col='category3', hue='category1', style='category2', kind="line", col_wrap=2, data=df, facet_kws=dict(legend_out=False))
h,l = plot.axes[0].get_legend_handles_labels()
plot.axes[0].legend_.remove()
plot.fig.legend(h,l, ncol=2) # you can specify any location parameter you want here

【讨论】：

谢谢，这有助于得到我想要的。我为我的问题添加了我的最终解决方案。

【参考方案2】：

感谢@DizietAsahi 的最终解决方案

import pandas as pd
import seaborn as sns

columns = ['category1', 'category2', 'category3', 'time', 'value']

data = [['content1', 'other1', 'critera1', 0, 0.1], ['content1', 'other1', 'critera1', 1, 0.4], ['content1', 'other1', 'critera1', 2, 0.7], ['content2', 'other1', 'critera1', 0, 0.2], ['content2', 'other1', 'critera1', 1, 0.6], ['content2', 'other1', 'critera1', 2, 0.8], ['content1', 'other2', 'critera1', 0, 0.0], ['content1', 'other2', 'critera1', 1, 0.2], ['content1', 'other2', 'critera1', 2, 0.8], ['content2', 'other2', 'critera1', 0, 0.3], ['content2', 'other2', 'critera1', 1, 0.6], ['content2', 'other2', 'critera1', 2, 0.5], [
    'content1', 'other1', 'critera2', 0, 0.1], ['content1', 'other1', 'critera2', 1, 0.4], ['content1', 'other1', 'critera2', 2, 0.7], ['content2', 'other1', 'critera2', 0, 0.2], ['content2', 'other1', 'critera2', 1, 0.6], ['content2', 'other1', 'critera2', 2, 0.8], ['content1', 'other2', 'critera2', 0, 0.0], ['content1', 'other2', 'critera2', 1, 0.2], ['content1', 'other2', 'critera2', 2, 0.8], ['content2', 'other2', 'critera2', 0, 0.3], ['content2', 'other2', 'critera2', 1, 0.6], ['content2', 'other2', 'critera2', 2, 0.5], ]

df = pd.DataFrame(data, columns=columns)

plot = sns.relplot(x='time', y='value', col='category3', hue='category1', style='category2', kind="line",
                   col_wrap=2, data=df)

handles, labels = plot.axes[0].get_legend_handles_labels()
plot._legend.remove()
plot.fig.legend(handles, labels, ncol=2, loc='upper center', 
                bbox_to_anchor=(0.5, 1.15), frameon=False)

【参考方案3】：

使用ncol

ax = sns.barplot(x="X", y="Y", data=data)    
ax.legend(loc='upper left',ncol=2, title="Title")

【参考方案4】：

@DizietAsahi 给出的答案不适用于我的情况。但是，我可以使用 sns.move_legend() 实用函数来做到这一点，它可用于修改绘图的图例。

sns.move_legend(ax, "upper center", bbox_to_anchor=(0.5, 1.15), 
                ncol=2)