在 Pandas 中,使用 .reset_index()
方法可以重置 .groupby()
输出中的索引。
.groupby()
方法用于按照指定的列或多个列对数据进行分组,并返回一个 GroupBy 对象。在进行分组操作后,通常会对分组结果进行聚合操作,例如计算平均值、求和等。聚合操作完成后,结果的索引会变为分组的键。
如果想要将分组结果的索引恢复为默认的整数索引,可以使用 .reset_index()
方法。该方法会将索引列转换为普通的列,并重新生成默认的整数索引。
以下是一个示例代码:
import pandas as pd
# 创建示例数据
data = {
'Name': ['Alice', 'Bob', 'Charlie', 'Alice', 'Bob'],
'Age': [25, 30, 35, 40, 45],
'Salary': [5000, 6000, 7000, 8000, 9000]
}
df = pd.DataFrame(data)
# 按照 Name 列进行分组,并计算平均薪资
grouped = df.groupby('Name')['Salary'].mean()
# 重置索引
reset_index = grouped.reset_index()
print(reset_index)
输出结果如下:
Name Salary
0 Alice 6500
1 Bob 7500
2 Charlie 7000
在这个例子中,我们首先按照 Name
列对数据进行分组,并计算了每个人的平均薪资。然后使用 .reset_index()
方法重置了索引,将 Name
列恢复为普通的列,并重新生成了默认的整数索引。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云