当省略x或y参数时,swarmlot无法识别色调

2022-02-27 00:00:00 python pandas matplotlib seaborn

问题描述

data = {'name': ['A', 'B', 'C', 'D'],
        'score': [-9.5, -8.3, -8.1, -7.0],
        'color': [4, 3, 2, 1]}

df = pd.DataFrame(data)
我将我的数据放在类似于上面的数据框中,并且我正在将其绘制到如下所示的海运沼泽图中。这些点是根据它们的分数绘制的,根据它如何落在3条虚线之间,我想给这些点加上不同的颜色。我使用df的‘color’列根据‘Score’值所在的位置分配一个键,该键对应于颜色和字典。

colors = {1:'pink', 2:'orange', 3:'red', 4:'green'}

然后我使用以下代码创建swarmlot,并将颜色字典映射到我的df的Colors列。

ax = sns.swarmplot(data=df, y='score', s=10, c=df['color'].map(colors))

当我这样做时,我不会生成任何错误,但是没有应用任何颜色,并且这些点保持其默认的蓝色(下图,左图)。那么,如何根据我的df['color']列为海运地块中的点指定颜色?

最后注意:当我尝试使用palette=df['color'].map(colors)而不是c=df['color'].map(colors)时,图形只会将所有内容更改为colors词典中的最后一个颜色(下图,右侧)

编辑:感谢您的建议@Trenton McKinney。这在某种程度上是成功的,因为颜色确实正确映射,但是当我包括x的‘name’列(它的实际标题)时,如下所示,我的点像散点图一样绘制,而不是像群图一样绘制。但如果尝试从参数中删除x='Title',则会出错。

ax = sns.swarmplot(data=data, x='Title', y='score', s=10, hue='color', palette=colors)


解决方案

  • 根据seaborn issue 941,这是预期行为。
  • 如果只指定x或只指定y,接口似乎玩得不好。
    • 根据数据帧的长度传递相同字符串的列表即可解决此问题:['']*len(df)['text']*len(df)
ax = sns.swarmplot(data=df, x=['']*len(df), y='score', hue='color', palette=colors)

相关文章