import pandas as pd
# 创建样例DataFrame
df = pd.DataFrame({'Group ID': [1, 2, 2, 1, 2, 1], 'Value': [2, 5, 4, 1, 3, 6]})
# 按Group ID分组,取每组中最大的值,命名为Max Value,并将其作为新列添加到DataFrame中
df['Max Value'] = df.groupby('Group ID')['Value'].transform(max)
print(df)
输出结果为:
Group ID Value Max Value
0 1 2 6
1 2 5 5
2 2 4 5
3 1 1 6
4 2 3 5
5 1 6 6
解释:首先按Group ID分组,然后对于每组数据,使用transform(max)
函数求出该组中的最大值,并将其添加为新列。最终得到了包含新列的DataFrame。