Matplotlib / Seaborn图例在添加标签后更改样式

时间:2019-05-23 17:53:50

标签: python pandas matplotlib seaborn

我正在使用seaborn从pandas df绘制一些数据。使用以下代码,几乎所有内容都能很好地绘制:

import pandas as pd
import seaborn as sns

sns.set(style='whitegrid', palette='muted')

legend = ["Hue 1", "Hue 2"]
order = ["A", "B"]

ax = sns.violinplot(x=df.xaxis, y=df.yaxis, hue=df.hue,
                    split=True, order=order)
ax.set_ylim(0, 100)
ax.set(xlabel='X - axis', ylabel='Y - axis')
ax.legend(title='Legend', loc='upper left', labels=legend)
ax.set_title('My little plot')
plt.show()

添加labels=后,图例中显示的“线型”就会发生变化。以下是屏幕截图。不幸的是,我的数据集太大而无法发布,所以我希望足够。

先谢谢了。 BBQuercus:)

左边不带,右边labels(R,C是我数据中的值)。

Without labels With labels

1 个答案:

答案 0 :(得分:1)

您可以尝试为图例绘制自定义补丁。我还没有测试过,但是我认为它应该可以工作。

from matplotlib.patches import Patch
palette=sns.color_palette('muted')

bluepatch = Patch(
    facecolor=palette[0],edgecolor='k',label='Hue 1'
)
orangepatch = Patch(
    facecolor=palette[1],edgecolor='k',label='Hue 2'
)

ax.legend(
    labels=['Hue 1','Hue 2'], 
    handles=[bluepatch, orangepatch], 
    title='Legend', 
    loc='upper left'
)