Python:创建带有条件和“包含”-Statement的新列

import pandas as pd

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

# 创建一个新列，根据条件设置值
df['Category'] = ''
df.loc[df['Age'] < 30, 'Category'] = 'Young'
df.loc[df['Age'] >= 30, 'Category'] = 'Old'

# 创建一个新列，判断元素是否存在于集合中
categories = ['Young', 'Middle-aged', 'Old']
df['IsCategory'] = df['Category'].isin(categories)

# 打印结果
print(df)

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