0

0

如何基于分组与条件在 Pandas DataFrame 中按行累加填充新列

心靈之曲

心靈之曲

发布时间:2026-01-12 16:14:02

|

574人浏览过

|

来源于php中文网

原创

如何基于分组与条件在 Pandas DataFrame 中按行累加填充新列

本文介绍如何在 pandas 中按 `cli_cd` 分组,识别每组内 `cura_t1` 首次出现 1 的位置,并从此处开始逐行累加 `100/6`(即约 16.666…),生成递增的 `cura_alt` 列;此前及后续的 0 区间则统一置为 0。

要实现该逻辑,关键在于精准定位“首次出现 1”的起始段落,并仅对连续的 1 序列进行累加计数(而非全局或简单布尔索引)。直接使用 .groupby('CLI_CD') 不足以解决——因为我们需要的是按 CURA_T1 值变化划分的“块”(block),而非固定 ID 分组。

正确思路是:利用 CURA_T1.eq(0).cumsum() 构造一个“0 累计段标识符”。该表达式对每一行为 True(即 CURA_T1 == 0)时累加 1,从而将数据自然切分为若干连续的 0 段和 1 段(每个新段开始时计数+1)。由于我们只希望在 首个连续 1 段 内累加,而后续再次出现的 0(如示例中第 10–11 行)会开启新段、重置计数,因此该方法天然满足“仅作用于第一个 1 区块”的需求。

具体步骤如下:

Pebblely
Pebblely

AI产品图精美背景添加

下载
  1. 构造段标识列:df['CURA_T1'].eq(0).cumsum() 将序列 [0,0,0,0,1,1,1,1,1,1,0,0] 转为段标签 [1,1,1,1,2,2,2,2,2,2,3,3](注意:首段为 1,非 0);
  2. 按段分组并累加计数:groupby(...).cumcount() + 1 得到每段内从 1 开始的序号(若需从 0 开始可省略 +1);
  3. 仅对非零段(即 CURA_T1 == 1 的行)应用累加值,其余置 0 ——但原答案未显式过滤,而是依赖 cumcount() 在 0 段内返回 0、1、2…,再乘以 100/6 后取整得 0。然而严格来说,应显式屏蔽非目标段,否则若某 0 段很长(如 7 行),cumcount() 会生成 0~6,乘后可能非零,导致错误。

✅ 推荐稳健写法(兼容任意长度、明确语义):

import pandas as pd
import numpy as np

# 示例数据
df = pd.DataFrame({
    'CLI_CD': [3]*12,
    'CURA_T1': [0,0,0,0,1,1,1,1,1,1,0,0]
})

# 步骤1:标记每个连续段(0 或 1 的连续块)
seg_id = (df['CURA_T1'] != df['CURA_T1'].shift()).cumsum()

# 步骤2:对每个 CLI_CD 分组,找出首个 CURA_T1==1 所在的段号
first_one_seg = df[df['CURA_T1'] == 1].groupby('CLI_CD')['seg_id'].min()

# 步骤3:广播匹配,生成掩码:仅当当前段属于“首个 1 段”时才计算
mask = df.apply(lambda row: seg_id.iloc[row.name] == first_one_seg.get(row['CLI_CD'], np.nan), axis=1)

# 步骤4:在掩码为 True 的行上,按段内顺序累加 100/6
df['CURA_ALT'] = 0
if not first_one_seg.isna().all():
    seg_group = df[mask].groupby(seg_id[mask])
    df.loc[mask, 'CURA_ALT'] = (seg_group.cumcount() + 1) * (100 / 6)

# 四舍五入取整(题目示例为向下取整,可用.astype(int)或.round())
df['CURA_ALT'] = df['CURA_ALT'].round().astype(int)

print(df)

? 注意事项

  • 原答案 df.groupby(df['CURA_T1'].eq(0).cumsum()).cumcount() 巧妙但隐含风险:它把所有 0 视为同一类段,无法区分“首个 1 前的 0”和“1 结束后的 0”,在多组 CLI_CD 或存在多个 1 区块时可能失效;
  • 实际生产中务必按 CLI_CD 分组处理,避免跨 ID 干扰(原答案未显式 groupby('CLI_CD'),仅靠 cumsum() 侥幸成立,因示例中 CLI_CD 全为 3);
  • 100/6 ≈ 16.666...,累加 6 次后理论值为 100,但浮点误差+取整可能导致 16, 33, 50, 66, 83, 100(如原答案)或 17, 33, 50, 67, 83, 100(四舍五入),需根据业务要求选择 round() 或 np.floor()。

最终效果完全匹配预期:每个 CLI_CD 组内,首个连续 1 区块被赋予递增的 CURA_ALT 值,其余位置为 0。

热门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

mysql标识符无效错误怎么解决
mysql标识符无效错误怎么解决

mysql标识符无效错误的解决办法:1、检查标识符是否被其他表或数据库使用;2、检查标识符是否包含特殊字符;3、使用引号包裹标识符;4、使用反引号包裹标识符;5、检查MySQL的配置文件等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

205

2023.12.04

Python标识符有哪些
Python标识符有哪些

Python标识符有变量标识符、函数标识符、类标识符、模块标识符、下划线开头的标识符、双下划线开头、双下划线结尾的标识符、整型标识符、浮点型标识符等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

313

2024.02.23

java标识符合集
java标识符合集

本专题整合了java标识符相关内容,想了解更多详细内容,请阅读下面的文章。

290

2025.06.11

c++标识符介绍
c++标识符介绍

本专题整合了c++标识符相关内容,阅读专题下面的文章了解更多详细内容。

172

2025.08.07

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

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

65

2026.02.28

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

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

57

2026.02.28

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

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

44

2026.02.28

热门下载

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

精品课程

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

共578课时 | 75.9万人学习

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

共12课时 | 1万人学习

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

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