Python/Pandas -基于多个变量和if/elif/else函数创建新变量

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [10, 20, 30, 40, 50],
        'C': [100, 200, 300, 400, 500]}

df = pd.DataFrame(data)

# 使用if/elif/else函数创建新变量
df['D'] = ''
df.loc[df['A'] > 3, 'D'] = '大于3'
df.loc[(df['A'] <= 3) & (df['B'] > 20), 'D'] = '小于等于3且B大于20'
df.loc[(df['A'] <= 3) & (df['B'] <= 20), 'D'] = '小于等于3且B小于等于20'

# 打印结果
print(df)

   A   B    C            D
0  1  10  100  小于等于3且B小于等于20
1  2  20  200  小于等于3且B小于等于20
2  3  30  300  小于等于3且B大于20
3  4  40  400          大于3
4  5  50  500          大于3