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

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 ) ) )

参考链接： blog.csdn/weixin_40161254/article/details/86231869