Pandas的匿名函数（lambda）中if elif elif的实现方法

（作者：陈玓玏）

Pandas的匿名函数lambda，能够避免写简短函数也需要定义的麻烦，配合apply的使用还能实现map功能，轻松对每一行使用同样的操作。

lambda中比较熟悉A if 条件 else B的用法，即满足条件则输出A，不满足则输出B，但if elif elif的用法或陌生些。

有两种方法可以实现。

一. 直接定义函数，用if elif elif

def f(x):
    if x==A:
       return 1
    elif x==B:
       return 2
    elif x==C:
       return 3
    else:
       return 4
df['col1'] = df['col2'].apply(lambda x:f(x))
           

二. 直接在表达式中写逻辑

df['col1'] = df['col2'].apply(lambda x:
                 1 if x==A
                    else (2 if x==B
                                else (3 if x==C
                                            else 4
                                        )
                           )
                  )
           

参考链接：

https://blog.csdn.net/weixin_40161254/article/details/86231869