Pandas在每个唯一列值的末尾添加合计行

import pandas as pd

# 假设数据存储在一个DataFrame中，命名为df
# 唯一列为'column_name'
# 待合计的列为'column_to_sum'

# 对数据进行分组，并计算每个分组的合计值
df_grouped = df.groupby('column_name').agg({'column_to_sum': 'sum'})

# 创建一个新的DataFrame，用于存储添加了合计行的结果
df_result = pd.DataFrame(columns=df.columns)

# 遍历原始数据DataFrame的每一行
for index, row in df.iterrows():
    # 将原始数据的每一行添加到结果DataFrame中
    df_result = df_result.append(row)
    
    # 检查当前行对应的唯一列值是否需要添加合计行
    if row['column_name'] in df_grouped.index:
        # 获取对应的合计值
        sum_value = df_grouped.loc[row['column_name'], 'column_to_sum']
        
        # 创建合计行，将合计值填入相应列
        sum_row = row.copy()
        sum_row['column_to_sum'] = sum_value
        
        # 添加合计行到结果DataFrame中
        df_result = df_result.append(sum_row)

# 打印添加合计行后的结果DataFrame
print(df_result)