从一个具有不同值和类型的列创建新的数据框列

import pandas as pd

# 创建示例数据框
data = {
    'Age': [25, 30, 35, 40, 45],
    'Gender': ['M', 'F', 'M', 'F', 'M']
}
df = pd.DataFrame(data)

# 定义年龄段函数
def get_age_group(age):
    if age < 30:
        return 'Young'
    elif age < 40:
        return 'Middle-aged'
    else:
        return 'Senior'

# 应用函数创建新列
df['AgeGroup'] = df['Age'].apply(get_age_group)

print(df)

   Age Gender  AgeGroup
0   25      M     Young
1   30      F  Middle-aged
2   35      M  Middle-aged
3   40      F     Senior
4   45      M     Senior

def get_age_group_safe(age):
    try:
        if age < 30:
            return 'Young'
        elif age < 40:
            return 'Middle-aged'
        else:
            return 'Senior'
    except TypeError:
        return None

df['AgeGroupSafe'] = df['Age'].apply(get_age_group_safe)