0

0

使用Pandas cummax 函数高效跟踪数据流中的累计最大值

碧海醫心

碧海醫心

发布时间:2025-11-20 12:21:11

|

524人浏览过

|

来源于php中文网

原创

使用pandas cummax 函数高效跟踪数据流中的累计最大值

本文详细介绍了如何在Pandas DataFrame中高效地创建一个新列,该列能够跟踪并保留数据流中遇到的累计最大值。通过利用Pandas内置的`cummax()`函数,可以简洁而优雅地解决当序列值增加时更新最大值,并在值下降时保持前一个最大值的需求,避免了复杂的迭代或分组逻辑。

需求概述:跟踪并保留累计最大值

在数据分析场景中,我们经常需要处理序列数据,并希望在一个新列中记录某个特定指标的“历史最高点”。具体来说,需求是这样的:对于DataFrame中的一列(例如a),我们希望创建一个新列c。c列的每个值应是到当前行为止,a列中出现过的最大值。这意味着,如果当前行的a值大于c列的当前值(即之前的累计最大值),那么c列就更新为这个新的最大值;如果当前行的a值小于或等于c列的当前值,c列则保持不变,继续保留之前的累计最大值。

让我们通过一个具体的DataFrame示例来阐明这一需求:

import pandas as pd

df = pd.DataFrame(
    {
        'a': [110, 115, 112, 180, 150, 175, 160, 145, 200, 205, 208, 203, 206, 207, 208, 209, 210, 215],
        'b': [1, 1, 0, 1, 0, 1, 0, 0, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1], # 'b'列在此问题中作为辅助理解,但并非解决核心问题的必要条件
    }
)

print("原始DataFrame:")
print(df)

期望得到的输出df['c']列如下:

      a  b    c
0   110  1  110
1   115  1  115
2   112  0  115 # a[2]=112 < c[1]=115, c保持115
3   180  1  180
4   150  0  180 # a[4]=150 < c[3]=180, c保持180
5   175  1  180 # a[5]=175 < c[4]=180, c保持180
6   160  0  180
7   145  0  180
8   200  1  200
9   205  1  205
10  208  1  208
11  203  0  208 # a[11]=203 < c[10]=208, c保持208
...

传统思路与潜在复杂性

初次面对这类问题时,开发者可能会倾向于考虑使用循环、条件判断,或者利用groupby结合一些自定义函数来识别“连续上涨”的“streak”并提取其中的最大值。例如,尝试通过df['b'].ne(df['b'].shift()).cumsum()来标识连续块,然后对这些块进行max操作。这种方法虽然在某些分组场景下有效,但对于单纯的“累计最大值”问题而言,会引入不必要的复杂性,并且可能效率较低。

简洁高效的解决方案:pandas.Series.cummax()

Pandas库为这类累计计算提供了非常强大的内置函数。针对“累计最大值”的需求,最直接且最高效的工具是pandas.Series.cummax()方法。

cummax()函数的作用是计算Series的累计最大值。它遍历Series中的每个元素,并返回一个新Series,其中每个位置的值是到当前位置为止,原始Series中出现过的最大值。这完美契合了我们的需求。

实现代码:

AIBox 一站式AI创作平台
AIBox 一站式AI创作平台

AIBox365一站式AI创作平台,支持ChatGPT、GPT4、Claue3、Gemini、Midjourney等国内外大模型

下载
# 使用cummax()函数创建列'c'
df['c'] = df['a'].cummax()

print("\n处理后的DataFrame:")
print(df)

输出结果:

原始DataFrame:
      a  b
0   110  1
1   115  1
2   112  0
3   180  1
4   150  0
5   175  1
6   160  0
7   145  0
8   200  1
9   205  1
10  208  1
11  203  0
12  206  1
13  207  1
14  208  1
15  209  1
16  210  1
17  215  1

处理后的DataFrame:
      a  b    c
0   110  1  110
1   115  1  115
2   112  0  115
3   180  1  180
4   150  0  180
5   175  1  180
6   160  0  180
7   145  0  180
8   200  1  200
9   205  1  205
10  208  1  208
11  203  0  208
12  206  1  208
13  207  1  208
14  208  1  208
15  209  1  209
16  210  1  210
17  215  1  215

从输出可以看出,df['c']列完全符合预期的行为,简洁而高效地实现了累计最大值的跟踪。

扩展:如何生成辅助列 b

原始问题中提到了一个辅助列b,其定义是当df.a > df.a.shift(1)时为1,否则为0。虽然在解决累计最大值问题时b列并非必需,但了解如何程序化地生成它也是一个很好的实践。

可以使用gt()(大于)方法结合shift()来比较当前行与上一行a的值,然后将布尔结果转换为整数:

# 生成列'b'的正确方法
df['b_generated'] = (df['a'] > df['a'].shift(1)).astype(int)

print("\n带有生成b列的DataFrame:")
print(df[['a', 'b', 'b_generated', 'c']])

输出示例:

带有生成b列的DataFrame:
      a  b  b_generated    c
0   110  1            0  110 # shift(1)对于第一行是NaN,比较结果为False
1   115  1            1  115
2   112  0            0  115
3   180  1            1  180
4   150  0            0  180
5   175  1            1  180
6   160  0            0  180
7   145  0            0  180
8   200  1            1  200
9   205  1            1  205
10  208  1            1  208
11  203  0            0  208
12  206  1            1  208
13  207  1            1  208
14  208  1            1  208
15  209  1            1  209
16  210  1            1  210
17  215  1            1  215

注意: 原始数据中的b列与根据规则生成的b_generated列在第一行有所不同。这是因为df.a.shift(1)在第一行会产生NaN,任何与NaN的比较(包括>)都会返回False。如果希望第一行b为1,可能需要对第一行进行特殊处理或调整逻辑。然而,对于本教程的核心问题——计算c列,b列的存在与否或其具体生成方式并不影响cummax()的正确应用。

总结

当需要在一个Pandas Series中跟踪并保留到当前点为止的累计最大值时,pandas.Series.cummax()函数是最佳选择。它提供了一个高度优化且易于理解的解决方案,避免了手动迭代或复杂的自定义函数,极大地提高了代码的简洁性和执行效率。掌握这类Pandas内置的累计函数(如cummin(), cumsum(), cumprod()等)对于高效处理时间序列或顺序数据至关重要。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

81

2025.12.04

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

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

33

2026.01.31

TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

25

2026.03.13

Python异步编程与Asyncio高并发应用实践
Python异步编程与Asyncio高并发应用实践

本专题围绕 Python 异步编程模型展开,深入讲解 Asyncio 框架的核心原理与应用实践。内容包括事件循环机制、协程任务调度、异步 IO 处理以及并发任务管理策略。通过构建高并发网络请求与异步数据处理案例,帮助开发者掌握 Python 在高并发场景中的高效开发方法,并提升系统资源利用率与整体运行性能。

44

2026.03.12

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

177

2026.03.11

Go高并发任务调度与Goroutine池化实践
Go高并发任务调度与Goroutine池化实践

本专题围绕 Go 语言在高并发任务处理场景中的实践展开,系统讲解 Goroutine 调度模型、Channel 通信机制以及并发控制策略。内容包括任务队列设计、Goroutine 池化管理、资源限制控制以及并发任务的性能优化方法。通过实际案例演示,帮助开发者构建稳定高效的 Go 并发任务处理系统,提高系统在高负载环境下的处理能力与稳定性。

50

2026.03.10

Kotlin Android模块化架构与组件化开发实践
Kotlin Android模块化架构与组件化开发实践

本专题围绕 Kotlin 在 Android 应用开发中的架构实践展开,重点讲解模块化设计与组件化开发的实现思路。内容包括项目模块拆分策略、公共组件封装、依赖管理优化、路由通信机制以及大型项目的工程化管理方法。通过真实项目案例分析,帮助开发者构建结构清晰、易扩展且维护成本低的 Android 应用架构体系,提升团队协作效率与项目迭代速度。

92

2026.03.09

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

102

2026.03.06

Rust内存安全机制与所有权模型深度实践
Rust内存安全机制与所有权模型深度实践

本专题围绕 Rust 语言核心特性展开,深入讲解所有权机制、借用规则、生命周期管理以及智能指针等关键概念。通过系统级开发案例,分析内存安全保障原理与零成本抽象优势,并结合并发场景讲解 Send 与 Sync 特性实现机制。帮助开发者真正理解 Rust 的设计哲学,掌握在高性能与安全性并重场景中的工程实践能力。

227

2026.03.05

热门下载

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

精品课程

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

共58课时 | 6万人学习

ASP 教程
ASP 教程

共34课时 | 5.9万人学习

Vue3.x 工具篇--十天技能课堂
Vue3.x 工具篇--十天技能课堂

共26课时 | 1.6万人学习

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

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