0

0

pandas 如何在 rolling/expanding 中使用自定义聚合函数

冷漠man

冷漠man

发布时间:2026-01-23 20:38:58

|

552人浏览过

|

来源于php中文网

原创

pandas的rolling/expanding自定义函数必须返回标量,返回Series/list会报错;需多输出时用apply+result_type='expand';expanding与rolling规则一致,仅窗口行为不同。

pandas 如何在 rolling/expanding 中使用自定义聚合函数

rolling/expanding 里传自定义函数必须返回标量

直接传 lambda x: x.mean() 没问题,但若函数返回 pd.Serieslist,会报 ValueError: Must produce aggregated value。pandas 的 rolling 和 expanding 要求聚合函数最终输出单个值(标量),不是数组、Series 或 DataFrame。

常见踩坑点:

  • .describe().quantile([0.25, 0.75]) 这类返回多个值的函数,会直接失败
  • 写成 lambda x: np.percentile(x, 90) 是 OK 的,因为返回 float;但 lambda x: np.percentile(x, [90, 95]) 就不行
  • 若真需要多输出(比如同时算 90% 和 95% 分位数),得包装成返回单个 object(如 tuple),再用 apply + result_type='expand' 拆开(见下一条)

想返回多个指标?用 apply + result_type='expand'

rolling(...).apply() 默认只接受标量返回,但加 result_type='expand' 后,允许函数返回 list/tuple/Series,pandas 会自动转成多列。

示例:窗口内同时计算 90% 和 95% 分位数

df['x'].rolling(5).apply(
    lambda x: np.quantile(x, [0.9, 0.95]), 
    result_type='expand'
).rename(columns={0: 'q90', 1: 'q95'})

注意:

  • result_type='expand' 仅在 apply 中有效,agg / aggregate 不支持
  • 返回的 tuple/list 长度必须固定,否则中间窗口长度不足时会出错(比如前 4 行无法满足 window=5,结果为 NaN,但 shape 仍按你定义的列数对齐)
  • 性能比单标量函数低,尤其数据量大时,避免在每行都做复杂计算

expanding 中用自定义函数和 rolling 完全一致

expanding()rolling() 共享同一套聚合逻辑,所有关于函数签名、返回值、result_type 的规则完全一样。区别只在窗口行为:rolling 是固定宽度滑动,expanding 是从首行累积增长。

所以以下写法是等价有效的:

Chromox
Chromox

Chromox是一款领先的AI在线生成平台,专为喜欢AI生成技术的爱好者制作的多种图像、视频生成方式的内容型工具平台。

下载
# 两种写法效果相同(假设 df['x'] 有足够长度)
df['x'].expanding(3).apply(lambda x: x.std(), engine='numba')  # 可加 numba 加速
df['x'].rolling(3).apply(lambda x: x.std())  # 但 rolling 前 2 行是 NaN

关键点:

  • min_periods 对两者都适用:expanding(min_periods=3) 表示前 2 行返回 NaN,第 3 行开始计算
  • 传入函数收到的 x 始终是 pd.Series(即使原列是 int64,x.dtype 也是对应类型),可放心调用 .to_numpy()np.array(x)
  • 别在函数里依赖全局变量或外部状态——pandas 可能并行或重排调用顺序

性能敏感时优先用 numpy 原生函数 + vectorize

自定义函数慢的主因是 Python 循环 + pandas Series 开销。如果逻辑能转成 numpy 向量化操作,应尽量避开 apply

例如计算滚动中位数偏移量:

# ❌ 慢:每窗口都调用 python median + sub
s.rolling(10).apply(lambda x: np.median(x) - x.iloc[-1])
<h1>✅ 快:先算好 rolling median(内置 C 实现),再减</h1><p>s.rolling(10).median() - s

更进一步:

  • numba.jit 编译简单数值函数(如自定义分位数、winsorize),配合 engine='numba'
  • 避免在 lambda 里重复创建对象(如每次 new list、pd.Series)
  • 确认是否真的需要自定义逻辑——很多需求其实已有内置方法:.corr().cov().kurt() 等都支持 rolling/expanding

实际用的时候,多数情况卡在“以为要自己写函数”,结果发现 pandas 已经内置了;少数真要定制的,核心就两条:返回标量、别碰外部状态。其余都是优化层面的事。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

80

2025.12.04

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

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

32

2026.01.31

css中float用法
css中float用法

css中float属性允许元素脱离文档流并沿其父元素边缘排列,用于创建并排列、对齐文本图像、浮动菜单边栏和重叠元素。想了解更多float的相关内容,可以阅读本专题下面的文章。

595

2024.04.28

C++中int、float和double的区别
C++中int、float和double的区别

本专题整合了c++中int和double的区别,阅读专题下面的文章了解更多详细内容。

108

2025.10.23

全局变量怎么定义
全局变量怎么定义

本专题整合了全局变量相关内容,阅读专题下面的文章了解更多详细内容。

95

2025.09.18

python 全局变量
python 全局变量

本专题整合了python中全局变量定义相关教程,阅读专题下面的文章了解更多详细内容。

106

2025.09.18

lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

215

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

192

2025.11.08

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

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

76

2026.03.11

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

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

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