使用group by键将DataFrame列转换为数组

import pandas as pd
import numpy as np

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

# 使用group by键将列转换为数组
result = df.groupby('Name').agg({'Age': lambda x: np.array(x), 'Salary': lambda x: np.array(x)})

# 输出结果
print(result)

             Age       Salary
Name                         
John    [30]         [7000]
Nick    [25, 40]     [6000, 9000]
Tom     [20, 35]     [5000, 8000]