pandas groupby 求和后,将聚合结果转换为 DataFrame
时间: 2024-05-12 09:16:37 浏览: 57
可以使用 `reset_index()` 方法将聚合结果转换为 DataFrame。
示例代码:
```python
import pandas as pd
# 创建数据
df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar',
'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'three',
'two', 'two', 'one', 'three'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]})
# 按 A 和 B 分组,求 C 和 D 的和
grouped = df.groupby(['A', 'B']).agg({'C': 'sum', 'D': 'sum'})
# 将聚合结果转换为 DataFrame
result = grouped.reset_index()
print(result)
```
输出结果:
```
A B C D
0 bar one 2 30
1 bar three 4 40
2 bar two 10 120
3 foo one 8 80
4 foo three 8 80
5 foo two 8 120
```
阅读全文