在Pandas中,可以使用groupby
方法将数据按照某个列进行分组,并对每个分组进行聚合操作。要将sum
值groupwise放入新列中,可以按照以下步骤进行操作:
import pandas as pd
groupby
方法按照某个列进行分组,例如按照group
列进行分组:grouped = df.groupby('group')
sum
值:sum_values = grouped['value'].sum()
map
方法将每个分组的sum
值映射到原始DataFrame中的对应行:df['sum_groupwise'] = df['group'].map(sum_values)
完整的代码示例:
import pandas as pd
# 创建DataFrame对象
data = {'group': ['A', 'A', 'B', 'B', 'B'],
'value': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
# 按照group列进行分组
grouped = df.groupby('group')
# 计算每个分组的sum值
sum_values = grouped['value'].sum()
# 将sum值添加到新列sum_groupwise中
df['sum_groupwise'] = df['group'].map(sum_values)
print(df)
输出结果:
group value sum_groupwise
0 A 1 3
1 A 2 3
2 B 3 12
3 B 4 12
4 B 5 12
在这个例子中,我们按照group
列进行分组,并计算每个分组的value
列的sum
值。然后,将每个分组的sum
值映射到原始DataFrame中的对应行,形成新的列sum_groupwise
。最后,输出包含新列的DataFrame。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云