使用hue绘制融化后DataFrame时Seaborn图例异常问题解析

云原生阿岚

2026-7-4

解决方案

问题根源

当hue参数对应字符串类型的离散变量时（比如融化后的variable列，值为'y1'/'y2'/'y3'），Seaborn会自动在图例中添加一个空的“占位”条目，用于处理数据中可能未出现的类别。而当hue是数值型离散变量时（比如对比代码中的group列，值为整数），不会生成这个额外条目。

直接调用ax.legend(labels=labels)时，原图例包含4个条目（1个空占位+3个实际类别），但你只传入了3个标签，导致标签和颜色错位，出现“偏移”现象。

解决方法

方法1：过滤空图例条目

获取原图例的handles，过滤掉空的占位条目，再重新设置标签：

fig, ax = plt.subplots(1,1)
sns.scatterplot(data=df_melted, x='x', y='value', hue='variable', ax=ax)
labels = ['a', 'b', 'c']

# 获取图例的handles，过滤第一个空条目
handles, _ = ax.get_legend_handles_labels()
ax.legend(title='my plot', handles=handles[1:], labels=labels)

plt.show()

方法2：重新创建图例

先移除原图例，再手动创建对应类别的图例条目：

fig, ax = plt.subplots(1,1)
sns.scatterplot(data=df_melted, x='x', y='value', hue='variable', ax=ax)
labels = ['a', 'b', 'c']

# 移除原图例
ax.get_legend().remove()

# 获取唯一类别和对应颜色
unique_vars = df_melted['variable'].unique()
colors = sns.color_palette()[:len(unique_vars)]

# 手动创建图例handle
handles = [plt.Line2D([], [], marker='o', color=colors[i], linestyle='', markersize=10) 
           for i in range(len(unique_vars))]

# 设置新图例
ax.legend(title='my plot', handles=handles, labels=labels)

plt.show()