使用类似regEx的switch-case向python DataFrame添加一个新列

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['John', 'Emma', 'Mike'],
        'Age': [25, 30, 35]}
df = pd.DataFrame(data)

# 定义一个函数，根据条件返回新列的值
def calculate_new_column(row):
    # 类似于switch-case的逻辑
    if row['Age'] < 30:
        return 'Young'
    elif row['Age'] < 40:
        return 'Middle-aged'
    else:
        return 'Old'

# 使用apply方法将函数应用于DataFrame的每一行，得到新列的值
df['Age Group'] = df.apply(calculate_new_column, axis=1)

# 输出结果
print(df)

   Name  Age    Age Group
0  John   25        Young
1  Emma   30  Middle-aged
2  Mike   35  Middle-aged