0

0

如何在 Pandas 中按分组标记首次满足条件的行

霞舞

霞舞

发布时间:2026-01-12 19:28:02

|

839人浏览过

|

来源于php中文网

原创

如何在 Pandas 中按分组标记首次满足条件的行

本文介绍如何基于分组(如客户名称)和布尔条件(如 `y/n` 列首次出现 `'y'`),为 dataframe 添加一列标记“首次满足条件的日期”,仅在对应行填充该日期,其余位置设为 `nan`。

在数据分析中,常需识别每个分组内首次满足特定条件的记录(例如:每位客户首次下单、首次响应“Y”、首次达标等),并将其关键字段(如日期)提取到新列中,其余行置空。Pandas 提供了高效向量化方法实现这一目标,无需循环或 apply。

核心思路分为三步:

  1. 构造基础布尔条件:筛选出所有 'Y/N' == 'Y' 的行;
  2. 按分组累计计数:对每组内满足条件的行进行累加计数,并标记“首次”(即累计值等于 1);
  3. 条件赋值:仅当某行同时满足 'Y' 且是本组首次时,才填入其 BAS_DT,否则为 NaN。

以下是完整实现代码:

import pandas as pd
import numpy as np

df = pd.DataFrame({
    'BAS_DT': ['2023-01-02', '2023-01-03', '2023-01-04', '2023-01-02', '2023-01-03'],
    'CUS_NAME': ['A', 'A', 'A', 'B', 'B'],
    'Y/N': ['Y', 'Y', 'Y', 'N', 'Y'],
    'cum_count': [1, 2, 3, 1, 2]
})

# 步骤 1:标记所有 'Y' 行
cond = df['Y/N'].eq('Y')

# 步骤 2:按 CUS_NAME 分组,对 cond 累计求和,并判断是否为组内首次(=1)
cond1 = cond.groupby(df['CUS_NAME']).cumsum().eq(1)

# 步骤 3:仅在同时满足 cond 和 cond1 的行填入 BAS_DT,否则为 NaN
df['occur_date'] = df['BAS_DT'].where(cond & cond1)

print(df)

输出结果与预期一致:

达奇AI论文写作
达奇AI论文写作

达奇AI论文辅助写作平台,在校学生、职场精英都在用的AI论文辅助写作平台

下载
       BAS_DT CUS_NAME Y/N  cum_count  occur_date
0  2023-01-02        A   Y          1  2023-01-02
1  2023-01-03        A   Y          2         NaN
2  2023-01-04        A   Y          3         NaN
3  2023-01-02        B   N          1         NaN
4  2023-01-03        B   Y          2  2023-01-03

关键技巧说明

  • df['col'].eq(val) 比 df['col'] == val 更推荐,语义清晰且支持链式调用;
  • groupby(...).cumsum() 在布尔 Series 上会自动转为整数累加(True→1, False→0),天然适合统计“第几次满足条件”;
  • where(condition) 是向量化赋值的优雅方式,比 np.where 或 loc 更简洁安全。

⚠️ 注意事项

  • 若某组内无任何 'Y'(如客户 'C' 全为 'N'),则 occur_date 全为 NaN,符合逻辑;
  • 数据顺序影响结果——此方法依赖原始行序,确保 BAS_DT 已按时间升序排列(如示例所示),否则“首次”可能不具业务意义;
  • 如需获取每个客户的首次发生日期作为独立汇总表,可改用 df[cond].groupby('CUS_NAME')['BAS_DT'].first()。

该方法兼具性能、可读性与扩展性,适用于各类“分组首现标记”场景。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
Python 时间序列分析与预测
Python 时间序列分析与预测

本专题专注讲解 Python 在时间序列数据处理与预测建模中的实战技巧,涵盖时间索引处理、周期性与趋势分解、平稳性检测、ARIMA/SARIMA 模型构建、预测误差评估,以及基于实际业务场景的时间序列项目实操,帮助学习者掌握从数据预处理到模型预测的完整时序分析能力。

76

2025.12.04

Python 数据清洗与预处理实战
Python 数据清洗与预处理实战

本专题系统讲解 Python 在数据清洗与预处理中的核心技术,包括使用 Pandas 进行缺失值处理、异常值检测、数据格式化、特征工程与数据转换,结合 NumPy 高效处理大规模数据。通过实战案例,帮助学习者掌握 如何处理混乱、不完整数据,为后续数据分析与机器学习模型训练打下坚实基础。

12

2026.01.31

Golang 测试体系与代码质量保障:工程级可靠性建设
Golang 测试体系与代码质量保障:工程级可靠性建设

Go语言测试体系与代码质量保障聚焦于构建工程级可靠性系统。本专题深入解析Go的测试工具链(如go test)、单元测试、集成测试及端到端测试实践,结合代码覆盖率分析、静态代码扫描(如go vet)和动态分析工具,建立全链路质量监控机制。通过自动化测试框架、持续集成(CI)流水线配置及代码审查规范,实现测试用例管理、缺陷追踪与质量门禁控制,确保代码健壮性与可维护性,为高可靠性工程系统提供质量保障。

43

2026.02.28

Golang 工程化架构设计:可维护与可演进系统构建
Golang 工程化架构设计:可维护与可演进系统构建

Go语言工程化架构设计专注于构建高可维护性、可演进的企业级系统。本专题深入探讨Go项目的目录结构设计、模块划分、依赖管理等核心架构原则,涵盖微服务架构、领域驱动设计(DDD)在Go中的实践应用。通过实战案例解析接口抽象、错误处理、配置管理、日志监控等关键工程化技术,帮助开发者掌握构建稳定、可扩展Go应用的最佳实践方法。

38

2026.02.28

Golang 性能分析与运行时机制:构建高性能程序
Golang 性能分析与运行时机制:构建高性能程序

Go语言以其高效的并发模型和优异的性能表现广泛应用于高并发、高性能场景。其运行时机制包括 Goroutine 调度、内存管理、垃圾回收等方面,深入理解这些机制有助于编写更高效稳定的程序。本专题将系统讲解 Golang 的性能分析工具使用、常见性能瓶颈定位及优化策略,并结合实际案例剖析 Go 程序的运行时行为,帮助开发者掌握构建高性能应用的关键技能。

35

2026.02.28

Golang 并发编程模型与工程实践:从语言特性到系统性能
Golang 并发编程模型与工程实践:从语言特性到系统性能

本专题系统讲解 Golang 并发编程模型,从语言级特性出发,深入理解 goroutine、channel 与调度机制。结合工程实践,分析并发设计模式、性能瓶颈与资源控制策略,帮助将并发能力有效转化为稳定、可扩展的系统性能优势。

20

2026.02.27

Golang 高级特性与最佳实践:提升代码艺术
Golang 高级特性与最佳实践:提升代码艺术

本专题深入剖析 Golang 的高级特性与工程级最佳实践,涵盖并发模型、内存管理、接口设计与错误处理策略。通过真实场景与代码对比,引导从“可运行”走向“高质量”,帮助构建高性能、可扩展、易维护的优雅 Go 代码体系。

18

2026.02.27

Golang 测试与调试专题:确保代码可靠性
Golang 测试与调试专题:确保代码可靠性

本专题聚焦 Golang 的测试与调试体系,系统讲解单元测试、表驱动测试、基准测试与覆盖率分析方法,并深入剖析调试工具与常见问题定位思路。通过实践示例,引导建立可验证、可回归的工程习惯,从而持续提升代码可靠性与可维护性。

3

2026.02.27

漫蛙app官网链接入口
漫蛙app官网链接入口

漫蛙App官网提供多条稳定入口,包括 https://manwa.me、https

235

2026.02.27

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Java 教程
Java 教程

共578课时 | 75.2万人学习

国外Web开发全栈课程全集
国外Web开发全栈课程全集

共12课时 | 1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号