将图例添加到由众多类别组成的barplot中

Question

这个问题与此one有关。在绘制具有不同颜色的条形图之后，我想添加一个相应的图例，但我不知道如何做。甚至有可能吗？

df = pd.DataFrame({'index':np.arange(0,10,1),'values':np.random.randint(0,100,10),
'group':list('aaabbbcccd')})

colors = {'a':'red','b':'gold','c':'lightblue','d':'cyan'}
df['values'].plot(kind='bar', color=[colors[i] for i in df['group']])

在当前解决方案中似乎不太可能，但是我尝试遍历组（使用groupby）并将其绘制到同一轴上，但是绘制不正确。还有其他建议吗？

Answer 1

您可以使用.pivot将事物放入正确的结构中，然后只需使用.plot：

df.pivot("index", "group")["values"].plot(kind="bar")