Python更新变量级别

Question

我有一个分类变量：df['reason']，有5个级别（0-4），我想将级别1和级别3分组到一个新的级别，称为'water'，级别2和级别4分组到'Non water'，不必理会级别0。所以我想在python中将它从5级减少到3级。我使用了for循环，但它没有返回正确的结果。谢谢你的意见

for i in df['reason']:
    if i==0:
        df['reason']='Not failed'
    elif i in [1,3]:
        df['reason']='Water'
    else:  
        df['reason']='Non-water'

Answer 1

level_map = {'level1': 'water', 'level3': 'water',
             'level2': 'Non-water', 'level4': 'Non-water', 
             'level5': 'level5'}

df['new_level'] = df['level'].apply(lambda x: level_map[x])