数据挖掘 - 如何在 Python 中使用 Groupby 计算累积和？ - 吾爱随笔录

我正在尝试使用 Pandas 的 DataFrame 计算 groupby 的累积总和。但是，我没有得到预期的输出。

我的源代码：

import pandas as pd
Employee = [['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
        ['CSE', 'CSE', 'EEE', 'EEE', 'CE', 'CE', 'ME', 'ME'],
        ['Cat-1', 'Cat-2', 'Cat-1', 'Cat-2', 'Cat-1', 'Cat-2', 'Cat-1', 'Cat-2']]

index = pd.MultiIndex.from_arrays(Employee, names=['Name', 'Dept', 'Category'])

Scale = [1, 2, 2, 3, 3, 1, 2, 3]
Salary = [100, 200, 200, 300, 300, 100, 200, 300]

df = pd.DataFrame({'scale': Scale,
               'salary': Salary},
              index=index)

df1 = df.groupby(['Category', 'scale']).cumsum()
print(df1)

预期输出：

Cat-1    1         100
         2         500
         3         800
Cat-2    1         100
         2         300
         3         900

获得的结果：

Name Dept Category        
A    CSE  Cat-1        100
B    CSE  Cat-2        200
C    EEE  Cat-1        300
D    EEE  Cat-2        500
E    CE   Cat-1        600
F    CE   Cat-2        600
G    ME   Cat-1       1000
H    ME   Cat-2       1200

Groupby 不起作用。但是，如果我使用sum()(ie df1 = df.groupby(['Category', 'scale']).sum()) 而不是cumsum()， groupby 可以完美运行。