groupby重设索引
时间: 2023-08-27 15:07:14 浏览: 87
在使用 `groupby` 方法之后,如果想要将分组后的结果重新设置为新的索引,可以使用 `reset_index()` 方法。
例如,假设有如下的数据集:
```
import pandas as pd
data = {'name': ['Alice', 'Bob', 'Charlie', 'David', 'Emma'],
'age': [25, 30, 35, 25, 40],
'gender': ['Female', 'Male', 'Male', 'Male', 'Female'],
'salary': [5000, 6000, 5500, 5000, 7000]}
df = pd.DataFrame(data)
```
现在,我们想要按照 `gender` 列进行分组,并计算每个分组中的平均工资:
```
grouped = df.groupby(['gender']).mean()
```
结果如下:
```
age salary
gender
Female 32.50 6000
Male 30.00 5500
```
现在,我们想要将 `gender` 列重新设置为新的索引,可以使用 `reset_index()` 方法:
```
grouped = grouped.reset_index()
```
结果如下:
```
gender age salary
0 Female 32.50 6000
1 Male 30.00 5500
```
可以看到,`reset_index()` 方法将 `gender` 列重新设置为了新的索引,并将原来的索引删除了。
阅读全文