Python pandas使用聚合创建条件列

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David', 'Eve'],
        'Age': [25, 30, 35, 40, 45],
        'Salary': [5000, 6000, 7000, 8000, 9000]}
df = pd.DataFrame(data)

# 使用聚合函数计算平均薪资
average_salary = df['Salary'].mean()

# 创建条件列，根据薪资是否高于平均薪资进行判断
df['Above Average'] = df['Salary'] > average_salary

# 打印结果
print(df)

      Name  Age  Salary  Above Average
0    Alice   25    5000          False
1      Bob   30    6000          False
2  Charlie   35    7000          False
3    David   40    8000           True
4      Eve   45    9000           True