pd groupby重命名列名
时间: 2023-08-24 07:07:58 浏览: 137
您可以使用`rename`方法来重命名`groupby`结果的列名。以下是一个示例:
```python
import pandas as pd
# 创建示例数据
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80}
df = pd.DataFrame(data)
# 使用groupby进行分组,并重命名列名
grouped = df.groupby(['A', 'B']).sum().reset_index()
grouped = grouped.rename(columns={'A': 'new_A', 'B': 'new_B'})
print(grouped)
```
输出:
```
new_A new_B C D
0 bar one 80 80
1 bar two 4 40
2 foo one 18 90
3 foo two 3 37
```
在上述示例中,我们使用`groupby(['A', 'B'])`对DataFrame进行分组,并使用`sum()`方法对分组结果进行求和。然后,我们使用`reset_index()`方法重置索引,并使用`rename()`方法将列'A'和'B'重命名为'new_A'和'new_B'。最终,我们打印出重命名后的分组结果。
阅读全文