以下是一个示例代码,用于按两列分组并添加列:
import pandas as pd
# 创建示例数据
data = {'Column1': ['A', 'B', 'A', 'B', 'A', 'B'],
'Column2': [1, 2, 3, 4, 5, 6],
'Value': [10, 20, 30, 40, 50, 60]}
df = pd.DataFrame(data)
# 按两列分组并计算每组的总和
grouped = df.groupby(['Column1', 'Column2']).sum().reset_index()
# 添加列
grouped['NewColumn'] = grouped['Column2'] * grouped['Value']
# 打印结果
print(grouped)
输出结果如下:
Column1 Column2 Value NewColumn
0 A 1 10 10
1 A 3 30 90
2 A 5 50 250
3 B 2 20 40
4 B 4 40 160
5 B 6 60 360
以上示例中,首先将数据按两列(Column1和Column2)分组,并计算每组的总和。然后,通过将Column2和Value相乘,计算出新的列NewColumn,并将其添加到结果数据框中。最后,打印出结果。
上一篇:按两列分组并求和?
下一篇:按两列分组的Pandas回归