在Pandas中,可以使用内置的聚合函数来避免在每一行上进行自定义聚合操作。以下是一个示例代码:
import pandas as pd
# 创建示例数据
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
'Age': [28, 32, 25, 28, 30],
'Salary': [5000, 6000, 4500, 5000, 5500]}
df = pd.DataFrame(data)
# 使用groupby和内置聚合函数进行聚合
agg_df = df.groupby('Name').agg({'Age': 'mean', 'Salary': 'sum'}).reset_index()
print(agg_df)
输出结果为:
Name Age Salary
0 John 27.5 10000
1 Nick 32.0 6000
2 Tom 28.0 10000
在上面的示例中,我们使用groupby
函数将数据按照Name
列进行分组。然后,使用agg
函数指定每一列需要进行的聚合操作,例如计算平均值mean
和求和sum
。最后,使用reset_index
函数将索引重置,并将结果保存到agg_df
中。这样就可以避免在每一行上进行自定义聚合操作。