pandas中新增的case_when方法

数据科技有技术 2024-05-24 11:25:36

来自:Python大数据分析 费弗里

1 简介

大家好我是费老师,pandas在前不久更新的2.2版本中,针对Series对象新增了case_when()方法,用于实现类似SQL中经典的CASE WHEN语句功能,今天的文章中,我们就来get其具体使用方法~

2 pandas中的case_when()新方法

首先请确保你的pandas版本大于等于2.2,在确保Python版本大于等于3.9的前提下,终端执行下列命令安装最新版本的pandas:

pip install pandas -U2.1 case_when()的使用

case_when()作为Series对象的方法,其参数非常简单只有一个caselist,用于定义条件映射规则,格式如[(条件1, 替代值1), (条件2, 替代值2), ...],最基础的用法下,每个条件为与目标Series长度相等的bool值序列,譬如下面的例子:

更灵活的方式,是将条件写作可执行函数,譬如lambda函数,进而引用自身实现灵活的条件判断:

函数式条件,在针对数据框进行「链式分析」的过程中,可以很灵活的基于上一步的「临时计算状态」,进行条件赋值操作,譬如(示例数据及代码见文章开头仓库地址):

0 阅读:2

数据科技有技术

简介:感谢大家的关注