python - 具有不等组的 Pandas 条形图

Question

我有具有层次结构的数据，想创建一个包含多组条形图的图。

import pandas as pd

data = [
 ['alpha', 'x', 1],
 ['alpha', 'y', 2],
 ['alpha', 'z', 2],
 ['beta', 'x', 3],
 ['beta', 'z', 4]]

df = pd.DataFrame(data, columns=['P','Q','R'])
df.pivot('P','Q','R').plot.bar(rot=0)

此代码产生:

我怎么能:

1. 消除缺失条的空间，即适应具有不同条数的组？
2. 让所有的 alpha 蓝色和 beta 橙色，即按组而不是在组内循环颜色？

Answer 1

如果您“手动”创建情节怎么办？您可以使用 loc 进行过滤。然后绘制在同一个图形上。

空间的神奇之处在于使用索引值。请注意，在 beta 中，我将 +1 添加到索引以创建额外的空间。然后，我将两个索引合并到 xticks 中，然后简单地使用 df['Q'] 作为标签。

plt.bar(data=df.loc[df['P']=='alpha'], x=df.loc[df['P']=='alpha'].index, height='R', label='alpha')
plt.bar(data=df.loc[df['P']=='beta'], x=df.loc[df['P']=='beta'].index+1, height='R', label='beta')
plt.xticks(df.loc[df['P']=='alpha'].index.tolist() + list(df.loc[df['P']=='beta'].index+1),df['Q'].tolist())

plt.legend()