当省略x或y参数时,swarmlot无法识别色调
问题描述
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)
相关文章