Python:将小计行添加到具有条件的数据框中

Python: Add subtotal row to a dataframe with conditions

我想在这里寻求一些菜鸟的帮助。我有一个带小计的旋转 DF,我想向它附加一个新的小计,它是其他一些小计的总和。例如,我想创建一个名为“DS”的新小计,即 CA-1、INT-1 和 INT-2 等其他小计的总和。在下图中,我附上了目标结果。

提前致谢!

Pivoted DF. Painted green aimed result

我已经弄明白了。假设你有 pandas pivot table 数据框,你想根据一些石斑鱼添加小计。让“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()