带有空列的堆积条形图答案

【问题标题】：Stacked Bar Graph with empty columns带有空列的堆积条形图
【发布时间】：2019-04-18 13:45:22
【问题描述】：

我正在尝试制作堆积条形图。我可以做一个基本的条形图：

df = pd.DataFrame({'Y': [1,1,1,1,1,2,3,2],
                   'X': [2,2,2,2,3,3,3,4]})

Y_1 = df.loc[df['Y'] == 1]
Y_2 = df.loc[df['Y'] == 2]

Count_0 = df.groupby(['X']).size().to_frame('Count').reset_index()
Count_1 = Y_1.groupby(['X']).size().to_frame('Count').reset_index()
Count_2 = Y_2.groupby(['X']).size().to_frame('Count').reset_index()

height_0 = Count_0.Count
height_1 = Count_1.Count
height_2 = Count_2.Count
bars     = Count_0.X

fig, (ax1) = plt.subplots(1,1);

y_pos = np.arange(len(bars))

p1 = plt.bar(y_pos, height_0) 

for item in ([ax1.title, ax1.xaxis.label, ax1.yaxis.label] +
             ax1.get_xticklabels() + ax1.get_yticklabels()):
    item.set_fontsize(22)

plt.xlabel('X')
plt.ylabel('Count')
plt.xticks(y_pos, bars)
plt.yticks(np.arange(0, 4.1, 1))
fig = plt.gcf()
fig.set_size_inches(18.5, 10.5)
plt.show()
plt.clf()

但是当我尝试按“Y”类堆叠它时：

p2 = plt.bar(y_pos, height_2, bottom = height_1)

我明白了：

ValueError: incompatible sizes: argument 'height' must be length 3 or scalar

我认为问题可能在于存在 Y = 2 和 Y = 3 的空列，因为这些类没有任何 X = 2 的实例。我希望 X 轴上的 X 和 Y 是颜色！

【问题讨论】：

标签： pandas matplotlib

【解决方案1】：

IIUC，你想要这个：

df = pd.DataFrame({'Y': [1,1,1,1,1,2,3,2],
                   'X': [2,2,2,2,3,3,3,4]})
df.groupby(['X','Y'])['Y'].count().unstack().plot.bar(stacked=True)

输出：

【讨论】：

【解决方案2】：

如果您可以使用不同的 Python 绘图库执行此操作，那么这里有一种使用 Altair 的堆积条形图的方法 - 否需要 groupby

导入和设置

import altair as alt
alt.renderers.enable('notebook')

堆积条形图

alt.Chart(df).mark_bar().encode(
    alt.X('X:N', axis=alt.Axis(labelAngle=0, tickSize=10)),
    alt.Y('count(Y):Q', axis=alt.Axis(title='Total count')),
    color='Y:N'
).properties(
    width=350,
    height=350
).configure_axis(
    titleFontSize=14,
    labelFontSize=12
).configure_legend(
    titleFontSize=14,
    labelFontSize=12
)

输出

这里是customize的链接

axis
legend

初步尝试

由于对 OP 问题的错误解释而被删除。

【讨论】：