news 2026/9/3 0:07:14

pandas实现sql的case when查询,原来这么简单!

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
pandas实现sql的case when查询,原来这么简单!

sql中的case when的功能是实现单列或者多列的条件统计,其实Pandas也可以实现的,比如万能的apply方法,就是写起来复杂一些,没有sql case when那么直观。

apply方法可以对dataframe、series执行特定函数,其实很强大,因为python什么逻辑都可以写。

举个例子,一张考试成绩的表scores,有语文和数学的得分,现在给考生综合打分,两门都在90以上打A,都在80-90之间打B,其他则打C。

这里如果用sql来查询的话,使用case when就很简单明了。

SELECT *, CASE WHEN chinese_score >= 90AND math_score >= 90THEN'A' WHEN chinese_score >= 80AND math_score >= 80THEN'B' ELSE'C' ENDAS score_type FROM scores

如果用pandas apply方法来实现的话,需要自定义一个判断函数,用来对成绩进行分类。

然后用apply方法应用到dataframe上,以下是完整代码,可以放到本地电脑跑。

import pandas as pd # 示例数据 data = { 'chinese_score': [90, 80, 79, 100, 89], 'math_score': [91, 95, 79, 99, 89], } df = pd.DataFrame(data) # 定义条件判断函数 def check_conditions(row): # 条件1:年龄>30,分数>80,性别为男性 result = 'A'if (row['chinese_score'] >= 90) and (row['math_score'] >= 90) \ else'B'if (row['chinese_score'] >= 80) and (row['math_score'] >= 80) \ else'C' return result # 应用apply生成标记列 score_type_1 = df.apply(check_conditions, axis=1) score_type_1

逻辑也很简单,判断函数实现多列的判断条件,apply将该判断函数应用到数据集上,就能实现类似sql case when的功能。

apply的应用场景很多,远比你想象的多得多,比如一张学生成绩信息的表,它可以给每位学生通过deepseek制定个性化的提升建议。

除了pandas apply能实现case when的功能外,numpy的select方法也能搞定,而且更为通用、简洁,建议试试。

import numpy as np import pandas as pd # 示例数据 data = { 'chinese_score': [90, 80, 79, 100, 89], 'math_score': [91, 95, 79, 99, 89], } df = pd.DataFrame(data) # 定义条件和对应的值 conditions = [ (df['chinese_score'] >= 90) & (df['math_score'] >= 90), (df['chinese_score'] >= 80) & (df['math_score'] >= 80) ] choices = ['A', 'B'] # select方法 score_type_1 = np.select(conditions, choices, default='C') score_type_1

numpy select方法类似case when语法,通过一个多列条件判断,区分不同的分类。

除了这两种,其实pandas种还有不少能实现的途径,但没必要尝试,因为这两种相对通用便捷,且符合python哲学。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/9/2 22:49:02

吐血整理,性能测试-TPS和事务响应时间关系详情,一篇通透...

目录:导读 前言一、Python编程入门到精通二、接口自动化项目实战三、Web自动化项目实战四、App自动化项目实战五、一线大厂简历六、测试开发DevOps体系七、常用自动化测试工具八、JMeter性能测试九、总结(尾部小惊喜) 前言 1、TPS和事务响应…

作者头像 李华
网站建设 2026/9/2 22:48:12

Java毕设选题推荐:springboot基于顾客偏好的唯品会推荐系统设计与实现基于SpringBoot+协同过滤的电商商品推荐系统【附源码、mysql、文档、调试+代码讲解+全bao等】

博主介绍:✌️码农一枚 ,专注于大学生项目实战开发、讲解和毕业🚢文撰写修改等。全栈领域优质创作者,博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围:&am…

作者头像 李华
网站建设 2026/9/2 22:48:15

DanceGRPO+FLUX:多模态生成强化学习模型的高效

一、背景介绍 Flux 模型:小模型高效生成高质量图像的基础 Flux 虽是百亿级参数的大模型家族,但其中的轻量化变体(如 Flux.1 (schnell))以及核心技术,为小尺寸模型提供了高效生成的范式。其关键技术优势适配小模型的优…

作者头像 李华
网站建设 2026/8/28 0:07:07

1.2 python及pycharm的安装

文章目录一、 下载安装包1.1 Python安装包1.2 PyCharm安装包1.3 PyCharm的设置二. 常见的问题及解决办法4.1 PyCharm无法启动4.2 PyCharm无法识别Python解释器4.3 PyCharm运行脚本时出错一、 下载安装包 1.1 Python安装包 从Python官方网站(https://www.python.or…

作者头像 李华
网站建设 2026/8/24 14:47:14

Leetcode 84 水果成篮 | 删除子数组的最大得分

1 题目 904. 水果成篮 你正在探访一家农场,农场从左到右种植了一排果树。这些树用一个整数数组 fruits 表示,其中 fruits[i] 是第 i 棵树上的水果 种类 。 你想要尽可能多地收集水果。然而,农场的主人设定了一些严格的规矩,你必…

作者头像 李华
网站建设 2026/9/2 7:20:53

基于微信小程序的零工市场服务系统计算机毕业设计项目源码文档

项目整体介绍在灵活就业场景多元化、零工供需匹配低效的背景下,传统零工市场存在 “信息分散、交易无担保、服务难监管” 的痛点,基于微信小程序构建的零工市场服务系统,适配零工求职者、用工方、平台管理员等角色,实现岗位发布、…

作者头像 李华