Pandas对多列使用条件判断新增列方式
作者:羊肉串儿加点辣
用户通过借鉴贴吧提问和回复,利用Pandas的df.apply结合lambda函数,根据子公司(A/B/C)及销售额≥1000的条件,新增"子公司规模"列,实现数据筛选与标记功能
想对几列进行条件判断,派生出新的一列,借鉴了百度贴吧楼主【南京的天】的提问,以及【nansky123456】、【ynwedgvt11】大佬的回复。
1.目标
我们有一个dataframe:
其中A、B、C是子公司,D、E是分公司,现在要新增一列‘子公司规模’,条件是:子公司而且销售额大于等于1000 ,达到如下效果:
2.方法
2.1 df.apply + lambda
2.2 df自身的判断
- 条件1:属于ABC已经知道是子公司。
- 条件2:属于子公司销售额大于等于1000。
只要满足以上这两个条件就可以了,加上本身pandas标签定位,根据条件返回的bool类型,
那么,只需两三行代码搞定,详情如下:
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。