python - 条形图图例基于按组而不是值对条形进行着色

Question

我已经按照描述创建了一个条形图 here我有多个变量(在“值”列中指示)并且它们属于重复组。我根据组成员身份对条形进行了着色。

我想创建一个最终相​​当于颜色字典的图例，显示与给定组成员身份相对应的颜色。

代码在这里:

d = {'value': [1, 2, 4, 5, 7 ,10], 'group': [1, 2, 3, 2, 2, 3]}
df = pd.DataFrame(data=d)
colors = {1: 'r', 2: 'b', 3: 'g'}
df['value'].plot(kind='bar', color=[colors[i] for i in df['group']])
plt.legend(df['group'])

这样，我只得到一个具有一种颜色 (1) 而不是 (1, 2, 3) 的图例。

谢谢!

Answer 1

您可以使用sns:

sns.barplot(data=df, x=df.index, y='value', 
            hue='group', palette=colors, dodge=False)

输出: