在pandas中基于groupedby的列求和,如何保留索引

在 pandas 中，可以使用 groupby 函数对数据进行分组，然后使用 sum 函数对分组后的数据求和。要保留索引，可以在调用 sum 函数时传入参数 as_index=False。

例如，假设有一个名为 df 的数据框，包含两列："A" 列和 "B" 列。要按 "A" 列的值对数据框进行分组，并对 "B" 列求和，并保留索引，可以这样做：

df.groupby("A", as_index=False)["B"].sum()

这将返回一个新的数据框，其中包含两列："A" 列和 "B" 列的和。索引也将被保留。