Python/Pandas 运行总计

Question

我有以下 Pandas 数据框:

Date          BrokenRule    Cost
2014-01-01    abc           100
2014-01-02    xyz           50
2014-01-03    abc           40
2014-01-04    xyz           70
2014-01-05    xyz           10

我需要计算每个 BrokenRule 的运行 Cost 总计，然后将其绘制为折线图。

如果我要计算所有 BrokenRule 的运行总数，我会简单地使用 cumsum()。但是，我似乎无法找到一种方法来显示每个 BrokenRule 的运行总数。也许通过同时使用 cumsum() 和 groupby() 但我无法让它工作。

有什么建议吗？谢谢

Answer 1

代码

df.set_index("BrokenRule", append=True).unstack().fillna(0).cumsum()

结果

            Cost     BrokenRule   abc  xyzDate                 2014-01-01   100    02014-01-02   100   502014-01-03   140   502014-01-04   140  1202014-01-05   140  130

通过取消堆叠索引的新添加的 BrokenRule 级别，我们为该列的每个单独值创建了一个列。用零填充然后执行 cumsum(默认情况下按列)给出了预期的结果。