下面是一个示例代码,演示如何按UUID进行分组,并为同一UUID的多个条目创建多个列:
import pandas as pd
# 创建示例数据
data = {
'UUID': ['a', 'a', 'b', 'b', 'c'],
'Value': [1, 2, 3, 4, 5]
}
df = pd.DataFrame(data)
# 按UUID进行分组
grouped = df.groupby('UUID')
# 创建多个列
df['Count'] = grouped.cumcount() + 1
df['Sum'] = grouped['Value'].transform('sum')
print(df)
输出结果:
UUID Value Count Sum
0 a 1 1 3
1 a 2 2 3
2 b 3 1 7
3 b 4 2 7
4 c 5 1 5
在这个示例中,我们首先创建了一个包含UUID和Value列的DataFrame。然后,我们使用groupby
方法将数据按UUID进行分组。接下来,我们使用cumcount
方法为每个分组创建一个计数列Count,从1开始计数。最后,我们使用transform
方法计算每个分组的Value列的总和,并将结果保存在Sum列中。
通过这种方式,我们可以对同一UUID的多个条目进行分组,并为它们创建多个列来保存不同的计算结果。