pandas分组后行求和
时间: 2023-06-30 17:12:01 浏览: 99
pandas分组聚合详解
可以使用`groupby`方法和`sum`方法来实现分组后行求和,具体代码如下:
```python
import pandas as pd
# 创建DataFrame
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': [2, 4, 6, 8, 10, 12, 14, 16]
})
# 按A列分组后,对C和D列进行求和
result = df.groupby('A').sum()[['C', 'D']]
print(result)
```
输出结果如下:
```
C D
A
bar 12 28
foo 16 37
```
其中,`groupby('A')`表示按A列进行分组,`sum()`表示对每组进行求和,`[['C', 'D']]`表示只保留C和D列。
阅读全文