(作者:陈玓玏)
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/weixin_40161254/article/details/86231869
更多推荐
Pandas的匿名函数(lambda)中if elif elif的实现方法
发布评论