避免在Pandas中遍历每一行进行自定义聚合。_编程开发

避免在Pandas中遍历每一行进行自定义聚合。

创始人

2024-12-17 10:01:20

0次

在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中。这样就可以避免在每一行上进行自定义聚合操作。

上一篇：避免在Pandas数据帧中使用循环遍历列值的方法

下一篇：避免在pandas中多次编写相同的.loc

避免在Pandas中遍历每一行进行自定义聚合。

相关内容

热门资讯