Python:向带条件的数据框添加小计行

Question

我想请一些新手来帮我。我有一个带有小计的旋转DF，我想向它追加一个新的小计，它是一些其他小计的总和。例如，我想创建一个名为"DS“的新小计，它是CA-1、INT-1和INT-2等其他小计的总和。在下面的图片中，我附加了目标结果。

提前感谢！

Pivoted DF. Painted green aimed result

Answer 1

我已经弄明白了。假设您有pandas数据透视表数据帧，并且您想要根据一些分组添加小计。假设"PIVOT“是pd的输出，您可以使用以下代码添加小计和总计：

PIVOT=pd.concat([PIVOT.assign(**{x: '' for x in ['index 1', 'index 2', 'index 3'][i:]}).groupby(list(['index 1', 'index 2', 'index 3'])).sum() for i in range(4)]).sort_index()