在DataFrame上使用groupedby函数时添加总计

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'Alice', 'Bob'],
        'Age': [25, 30, 35, 40, 45],
        'Salary': [5000, 6000, 7000, 8000, 9000]}
df = pd.DataFrame(data)

# 使用groupby函数按照Name列进行分组，并添加总计
grouped = df.groupby('Name').agg({'Age': 'sum', 'Salary': 'mean'})
total = df.agg({'Age': 'sum', 'Salary': 'mean'}).rename(index={'Age': 'Total', 'Salary': 'Total'})

# 将总计添加到分组结果中
grouped_with_total = pd.concat([grouped, total])

print(grouped_with_total)

         Age  Salary
Alice     65    6500
Bob       75    7500
Charlie   35    7000
Total    165    7200