pandas新列if语句

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)

# 使用if语句创建新的列
df['Category'] = ''
df.loc[df['Age'] < 30, 'Category'] = 'Young'
df.loc[df['Age'] >= 30, 'Category'] = 'Adult'

# 打印DataFrame
print(df)

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