在pandas中基于groupedby的列求和,如何保留索引
时间: 2023-01-11 22:55:36 浏览: 91
pandas 的str列运算.md
在 pandas 中,可以使用 `groupby` 函数对数据进行分组,然后使用 `sum` 函数对分组后的数据求和。要保留索引,可以在调用 `sum` 函数时传入参数 `as_index=False`。
例如,假设有一个名为 `df` 的数据框,包含两列:"A" 列和 "B" 列。要按 "A" 列的值对数据框进行分组,并对 "B" 列求和,并保留索引,可以这样做:
```
df.groupby("A", as_index=False)["B"].sum()
```
这将返回一个新的数据框,其中包含两列:"A" 列和 "B" 列的和。索引也将被保留。
阅读全文