在Python中,您可以使用pandas库来对聚合进行分组并包含单独的列。下面是一个示例代码:
import pandas as pd
# 创建一个示例数据集
data = {'Group': ['A', 'A', 'B', 'B', 'B', 'C'],
'Value': [1, 2, 3, 4, 5, 6],
'Extra': ['X', 'Y', 'Z', 'W', 'U', 'V']}
df = pd.DataFrame(data)
# 按Group进行分组并对Value列进行求和
grouped = df.groupby('Group')['Value'].sum()
# 将聚合结果与原始数据集合并
result = df.merge(grouped, left_on='Group', right_index=True, how='left')
print(result)
输出结果为:
Group Value Extra Value
0 A 1 X 3
1 A 2 Y 3
2 B 3 Z 12
3 B 4 W 12
4 B 5 U 12
5 C 6 V 6
在这个示例中,我们首先使用groupby
函数对Group
列进行分组,并对Value
列进行求和。然后,我们使用merge
函数将聚合结果与原始数据集合并,通过left_on
和right_index
参数指定按照Group
列进行合并。最后,我们打印出合并后的结果。