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 是从首行累积增长。

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

Kive
Kive

一站式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])

✅ 快:先算好 rolling median(内置 C 实现),再减

s.rolling(10).median() - s

更进一步:

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

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

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

772

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

662

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

765

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

679

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1385

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

570

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

579

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

751

2023.08.11

c++空格相关教程合集
c++空格相关教程合集

本专题整合了c++空格相关教程,阅读专题下面的文章了解更多详细内容。

0

2026.01.23

热门下载

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

精品课程

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

共4课时 | 15.3万人学习

Django 教程
Django 教程

共28课时 | 3.4万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.2万人学习

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

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