使用pandas创建新列时的条件拆分

import pandas as pd

# 创建一个示例DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'Gender': ['Female', 'Male', 'Male', 'Male']}
df = pd.DataFrame(data)

# 根据条件拆分数据并创建新列
df.loc[df['Age'] < 30, 'Age Group'] = 'Young'
df.loc[df['Age'] >= 30, 'Age Group'] = 'Old'

# 打印结果
print(df)

      Name  Age  Gender Age Group
0    Alice   25  Female     Young
1      Bob   30    Male       Old
2  Charlie   35    Male       Old
3    David   40    Male       Old