Pandas groupby分别制作两列列表

import pandas as pd

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

# 使用groupby函数按照Name列进行分组，并计算每个分组的平均年龄和平均工资
result = df.groupby('Name').agg({'Age': 'mean', 'Salary': 'mean'})

# 将结果转换为两列列表
result_list = result.reset_index().values.tolist()

print(result_list)

[['Alice', 25.0, 5000.0], ['Bob', 30.0, 6000.0], ['Charlie', 35.0, 7000.0]]