admin管理员组

文章数量:1794759

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

本文标签: 函数方法Pandaseliflambda