Pandas -基于其他列对列进行分组，并将它们标记到新列中

import pandas as pd

# 创建一个示例数据集
data = {
    'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
    'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
    'C': [1, 2, 3, 4, 5, 6, 7, 8]
}
df = pd.DataFrame(data)

# 使用groupby()函数进行分组，并将分组结果标记到新列'D'中
df['D'] = df.groupby(['A', 'B'])['C'].transform('sum')

# 打印结果
print(df)

     A    B  C   D
0  foo  one  1   9
1  bar  one  2   2
2  foo  two  3  15
3  bar  two  4   4
4  foo  two  5  15
5  bar  one  6   2
6  foo  two  7  15
7  foo  one  8   9