Python [panda/lambda] -按特定列对DF进行分组和聚合

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)

# 按Name列进行分组，并计算平均年龄和总薪资
grouped = df.groupby('Name')
result = grouped.agg({'Age': 'mean', 'Salary': 'sum'})

print(result)

         Age  Salary
Name                
Alice   32.5   13000
Bob     37.5   15000
Charlie 35.0    7000

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)

# 按Name列进行分组，并计算平均年龄和总薪资
result = df.groupby('Name').apply(lambda x: pd.Series({'Age': x['Age'].mean(), 'Salary': x['Salary'].sum()}))

print(result)