熊猫DataFrame累积值

Question

我有以下熊猫数据框：

>>> df

    Category     Year      Costs
0      A           1       20.00
1      A           2       30.00
2      A           3       40.00
3      B           1       15.00
4      B           2       25.00
5      B           3       35.00

如何添加一个累计成本列，该列累加了同一类别和以前年份的成本。 带有先前df的额外列的示例：

>>> new_df

    Category     Year      Costs    Cumulative Costs
0      A           1       20.00         20.00
1      A           2       30.00         50.00
2      A           3       40.00         90.00
3      B           1       15.00         15.00
4      B           2       25.00         40.00
5      B           3       35.00         75.00

建议？

Answer 1

这在pandas 0.17.0中有效。感谢@DSM在terser解决方案的注释中。

df['Cumulative Costs'] = df.groupby(['Category'])['Costs'].cumsum()

>>> df
  Category  Year  Costs  Cumulative Costs
0        A     1     20                20
1        A     2     30                50
2        A     3     40                90
3        B     1     15                15
4        B     2     25                40
5        B     3     35                75