使用分类数据的python堆积条形图

时间:2018-04-29 01:01:20

标签: python pandas matplotlib

我有一个看起来像这样的Pandas数据帧(1800 obs):

      A      B      C      D
 1   CL0    CL1    CL2    CL0
 2   CL2    CL1    CL1    CL3
 3   CL3    CL2    CL0    CL1
 .   ...    ...    ...    ...
 .   ...    ...    ...    ...
 n   CL2    CL1    CL0    CL3

我想创建一个堆积条形图,它在x轴上有'A','B','C','D'列,以及y轴上该要素中每个等级的百分比。如下图所示。

我猜我需要以某种方式将数据制成表格?但我不知道该怎么做。

enter image description here

1 个答案:

答案 0 :(得分:3)

print(df)

输出:

     A    B    C    D
1  CL0  CL1  CL2  CL0
2  CL2  CL1  CL1  CL3
3  CL3  CL2  CL0  CL1

使用.apply()

counts = df.apply(lambda x: x.value_counts() / len(x)).transpose()
fig = plt.figure()
ax = fig.add_subplot(111)
counts.plot(ax=ax,kind='bar', stacked=True, rot=0)
vals = ax.get_yticks()
ax.set_yticklabels(['{:3.2f}%'.format(x*100) for x in vals])
ax.yaxis.grid(True)
ax.set_axisbelow(True)
plt.show()

输出:

enter image description here