将堆叠表绘制成条形图 - Pandas

问题描述

我使用

将我的表格汇总到下表中
test_df = countData_df
stackedTable =  test_df.reset_index().pivot_table(values='volume',index=['address','direction'],aggfunc='sum')

enter image description here

不,我想创建一个条形图,在条形图上有四种不同方向的颜色。每个条形只能有两个选项:N 或 S; E 或 W

所以我完成了这条线

test_df.reset_index().pivot_table(values='volume',aggfunc='sum').plot(kind = 'bar',stacked = True,color = ['b','g','r','m'])

显示如下

enter image description here

解决方法

代码就快完成了。要修复情节,您需要解开方向。在示例中,方向是索引的一部分,但它应该是某种体积的分组元素。

test_df.reset_index().pivot_table(values='volume',index=['address','direction'],aggfunc='sum').unstack("direction").plot(kind = 'bar',stacked = True,color = ['b','g','r','m'])
,
df = df[['direction','volume']]
df = df.groupby('direction').sum()