如何在同一图形的多组垂直swarmplot中用数据集列设置hue？

问题 #

我使用swarmplot在同一图形中垂直展示多个分布，代码如下：

df = polars.read_csv("...")
ax = seaborn.swarmplot(data=[df["ColA"], df["ColB"], df["ColC"]])
ax.set_xticklabels(["A", "B", "C"])

但所有数据点颜色一致，尝试用数据集的ColD列作为hue参数（如hue=df["ColD"]）时，出现如下错误：

ValueError: The following variable cannot be assigned with wide-form data: `hue`

请问如何将数据集的一列作为图形的hue值？

解决方案 #

要实现按ColD列区分数据点颜色的蜂群图，需先将宽格式数据转换为长格式，具体代码如下：

df = polars.read_csv("...")
# 宽表转长表：以ColD为分组标识，将ColA/ColB/ColC转换为数值列
df_long = df.melt(id_vars='ColD', var_name='Measures', value_name='Score', value_vars=["ColA", "ColB", "ColC"])
# 基于长表绘制蜂群图，指定hue参数为ColD
ax = seaborn.swarmplot(data=df_long, hue='ColD', x='Measures', y='Score')
# 清除坐标轴默认标签
ax.set(xlabel='', ylabel='')
# 设置x轴刻度位置，避免警告
ax.set_xticks([i for i in range(3)])
# 自定义x轴刻度显示文本
ax.set_xticklabels(["A", "B", "C"])

内容的提问来源于stack exchange，提问作者FiReTiTi