绘制堆叠的数据框答案

【问题标题】：Plotting a stacked dataframe绘制堆叠的数据框
【发布时间】：2017-11-20 15:43:59
【问题描述】：

我想使用 pandas.DataFrame.plot.bar(stacked=True) 功能来可视化我的数据框（见下方标题）。

选择 1 个周期，我希望将每个“Wave”的“num”堆叠在“Electrodes”上。理想情况下，每个时期都将绘制在单独的子图中。当前代码：

one = df[df["Period"]=="1"]
one.plot.bar(stacked=True);
plt.show()

在下面的示例中，这意味着 a、b、c、d 代表不同的 Waves，0-9 代表不同的电极。

我认为这是将我的数据框重新排序到 plot.bar 的预期的问题，但我不确定如何继续。

这是 DataFrame 的前 30 行

    Electrode   Period  Wave    num
0   7   10  Beta    1
1   8   12  Beta    1
2   15  10  High gamma  1
3   4   10  Theta   1
4   11  4   High gamma  1
5   12  13  High gamma  3
6   11  4   Delta   3
7   11  4   Theta   0
8   14  0   Delta   2
9   14  1   Beta    0
10  11  6   Low gamma   1
11  1   9   Theta   0
12  8   1   Theta   0
13  5   8   Theta   0
14  10  0   Low gamma   2
15  13  12  Alpha   1
16  8   13  Alpha   1
17  10  0   Beta    1
18  7   5   Alpha   2
19  10  3   Theta   0
20  14  6   High gamma  2
21  4   11  Beta    1
22  4   5   Delta   1
23  4   10  High gamma  2
24  10  0   High gamma  3
25  12  4   Alpha   1
26  8   8   Theta   0
27  8   11  Beta    1
28  6   2   Delta   2
29  12  7   Low gamma   3

【问题讨论】：

我正在处理问题的第二部分，以前知道周期数吗？
它会随着组的变化而变化，但我们可以假设 6 个周期
好的，所以请测试我的解决方案，如果有问题，请告诉我。谢谢。

标签： python pandas dataframe plot stacked-chart

【解决方案1】：

我认为需要groupby 并将mean 与unstack 聚合以进行重塑：

one = df[df["Period"]==10].groupby(['Electrode','Wave'])['num'].mean().unstack()
one.plot.bar(stacked=True)

对于所有子图：

grouped = df.groupby('Period')
nrows = int(math.ceil(len(grouped)/2.))
fig, axs = plt.subplots(nrows,2, figsize=(12,30))

for (name, df), ax in zip(grouped, axs.flat):
    df.groupby(['Electrode','Wave'])['num'].mean().unstack().plot.bar(stacked=True, ax=ax)

【讨论】：

我收到此错误：“ValueError：索引包含重复条目，无法重塑”。我的“df.index”产生“RangeIndex(start=0, stop=1344, step=1)”

【解决方案2】：

根据 jezrael 启发的错误，我找到了这个解决方案：

df[df["Period"]=="1"].pivot_table(values="num", index=["Electrode"], columns="Wave", aggfunc='mean').plot.bar(stacked=True)

【讨论】：