如何编辑海洋传奇标题和图形功能的标签

Question

我已经使用seaborn和pandas dataframe（data）创建了此图：

我的代码：

import seaborn as sns

g = sns.lmplot('credibility', 'percentWatched', data=data, hue='millennial', markers=["+", "."])

您可能会注意到该情节的传奇标题只是变量名（'千年'）传说项目是其值（0，1）。如何编辑传奇的标题和标签？理想情况下，传奇标题将是“世代”，标签将是“千禧一代”和“老一辈”。

Answer 1

花了我一段时间来通读以上。这是我的答案：

import seaborn as sns
import matplotlib.pyplot as plt
tips = sns.load_dataset("tips")

g = sns.lmplot(
    x="total_bill", 
    y="tip", 
    hue="smoker", 
    data=tips,  
    legend=False
)

plt.legend(title='Smoker', loc='upper left', labels=['Hell Yeh', 'Nah Bruh'])
plt.show(g)

参考更多参数：

Answer 2

• 如果legend_out将设置为true，则可以通过g._legend属性获得Legend，它是图的一部分。 Seaborn Legend是标准Matplotlib传奇对象。因此，您可以更改传奇文本。
• 在python 3.8.11中测试，matplotlib 3.4.3，seaborn 0.11.2

import seaborn as sns

# load the tips dataset
tips = sns.load_dataset("tips")

# plot
g = sns.lmplot(x="total_bill", y="tip", hue="smoker", data=tips, markers=["o", "x"], facet_kws={'legend_out': True})

# title
new_title = 'My title'
g._legend.set_title(new_title)
# replace labels
new_labels = ['label 1', 'label 2']
for t, l in zip(g._legend.texts, new_labels):
    t.set_text(l)

如果legend_out设置为false，则另一种情况。您必须定义哪个轴具有传奇（在示例中为轴号0）：

g = sns.lmplot(x="total_bill", y="tip", hue="smoker", data=tips, markers=["o", "x"], facet_kws={'legend_out': False})

# check axes and find which is have legend
leg = g.axes.flat[0].get_legend()
new_title = 'My title'
leg.set_title(new_title)
new_labels = ['label 1', 'label 2']
for t, l in zip(leg.texts, new_labels):
    t.set_text(l)

您可以将两种情况结合起来并使用此代码：

g = sns.lmplot(x="total_bill", y="tip", hue="smoker", data=tips, markers=["o", "x"], facet_kws={'legend_out': True})

# check axes and find which is have legend
for ax in g.axes.flat:
    leg = g.axes.flat[0].get_legend()
    if not leg is None: break
# or legend may be on a figure
if leg is None: leg = g._legend

# change legend texts
new_title = 'My title'
leg.set_title(new_title)
new_labels = ['label 1', 'label 2']
for t, l in zip(leg.texts, new_labels):
    t.set_text(l)

​基于。

Answer 3

为了自定义传说，您需要直接操纵传说对象。 Seaborn中的传说存储在情节对象中。通过此传奇对象，您可以选择标题并修复标签的详细信息，以了解您的数据类型。

您可以做到这一点：

g = sns.lmplot('credibility', 'percentWatched', data=data, hue='millennial', markers=["+", "."])

g.set(title='Generation', xlabel='Credibility', ylabel='Percent Watched')
g._legend.set(title='Generation', labels=["Millennial", "Older Generations"])

plt.show()

Answer 4

以下是编辑海洋人物传说的其他一些方法（从Seaborn 0.13.2开始）。

1. 由于此处的传奇来自列传递给hue，这是最简单的方法（以及一种需要最少的工作IMO），如上所述在注释中将其用作hue变量。

    进口Seaborn作为SNS
   df = sns.load_dataset（“提示”）
   
   g = sns.lmplot（
       x ='total_bill'，y ='tip'， 
       data = df.Assign（性别= df ['sex']。地图（{'男性'：'男人'，'女性'：'woman'}）），＃添加一个新列
       色相='性别'，＃＆lt; ---使用新列作为色调
       标记= [“+”，“。”]
   ）
    

2. 另一种方法是隐藏默认的传说，并使用add_legend（）。

     g = sns.lmplot（x ='total_bill'，y ='tip'，data = df，hue ='sex ='sex'，markers = [“+” ，“。”]））
   g.legend.set_visible（false）＃隐藏原始传说（也可以将`legend = false“ false”传递给上面的绘图调用）
   
   ＃使用旧数据创建新的传奇数据
   映射= {'男性'：'男人'，'女性'：'女人'}
   leg_data = {映射[k]：v for K，v in G._legend_data.items（）}}
   ＃将新的传奇数据添加到图
   g.add_legend（legend_data = leg_data，title ='性别'，label_order = list（leg_data））
    

3. @serenity的答案效果很好，但是它不检查标签是否由正确的新标签替换。您可以使用IF-ELSE块进行操作，并确保用正确的标签替换标签。另外，您可以使用Legend而不是_legend。

     g = sns.lmplot（x ='total_bill'，y ='tip'，data = df，hue ='sex ='sex'，markers = [“+” ，“。”]））
   g.legend.set_title（“性别”）
   对于g.legend.texts中的标签：
       如果label.get_text（）==“男性”：
           label.set_text（“男人”）
       别的：
           label.set_text（“女人”）
    

上述所有选项都执行了编辑图例的以下转换。