Pandas DataFrame - 如何制作堆叠面积图堆栈（matplotlib）

Question

我正在尝试将 pandas DataFrame 中的数据转换为堆叠区域图，但似乎无法将其堆叠。

数据的格式

index | datetime (yyyy/mm/dd) | name | weight_change

每天测量 6 个不同的人。

我希望堆叠图显示日期时间 (x) 上的 weight_change (y)，但 6 个人中的每一个都显示 weight_change

我最接近它的是：

df = df.groupby['datetime', 'name'], as_index=False).agg({'weight_change': 'sum'})
agg = df.groupby('datetime').sum()
agg.plot.area()

这会生成 weight_change 值聚合的面积图（每天每个人 weight_change 的总和），但我不知道如何像这里的不同值一样为每个人拆分它：

我尝试了各种各样的事情，但都没有运气。有什么想法吗？

Answer 1

您的数据的简化版本：

df = pd.DataFrame(dict(days=range(4)*2,
                       change=np.random.rand(8)*2.,
                       name=['John',]*4 + ['Jane',]*4))

df:

    change  days    name
0   0.238336    0   John
1   0.293901    1   John
2   0.818119    2   John
3   1.567114    3   John
4   1.295725    0   Jane
5   0.592008    1   Jane
6   0.674388    2   Jane
7   1.763043    3   Jane

现在我们可以简单地使用pyplot的stackplot：

import matplotlib.pyplot as plt

days = df.days[df.name == 'John']
plt.stackplot(days, df.change[df.name == 'John'],
              df.change[df.name == 'Jane'])

这会产生以下情节：