0

0

如何在 Pandas 中对滚动窗口应用自定义函数并灵活返回任意列数的结果

心靈之曲

心靈之曲

发布时间:2026-01-30 18:54:09

|

424人浏览过

|

来源于php中文网

原创

如何在 Pandas 中对滚动窗口应用自定义函数并灵活返回任意列数的结果

本文介绍一种高效、无显式 for/while 循环的替代方案,利用 `numpy.sliding_window_view` 构建结构化滚动窗口,支持按列名操作 dataframe 子视图,并自由返回任意数量和名称的新列。

Pandas 原生的 DataFrame.rolling().apply() 在处理多列输入 → 多列输出(尤其是输出列数 ≠ 输入列数)且需按列名访问数据的场景时存在明显限制:raw=True 仅传入 NumPy 数组(丢失列名),而 raw=False 又强制要求返回值形状与输入一致(如 window.shape == (2, 3) 时必须返回 (2, 3))。这使得实现如“基于 A/B/C 列计算 D/E/F/G 四个新指标”的需求变得困难。

此时,推荐使用 NumPy 1.20+ 提供的 sliding_window_view —— 它能以零拷贝方式生成滑动窗口视图,再结合轻量级 pd.DataFrame 构造(设置 copy=False),即可在保持高性能的同时获得完整 DataFrame 接口:

from numpy.lib.stride_tricks import sliding_window_view
import pandas as pd
import numpy as np

# 示例数据
df = pd.DataFrame({
    "A": range(10),
    "B": range(10, 20),
    "C": range(20, 30)
})
cols = ["A", "B", "C"]
window_size = 2

# 预分配结果列表(首行为 NaN 占位,对应窗口未就绪行)
results = [tuple([np.nan] * 4)]  # 对应 D, E, F, G 四列

# 滑动窗口遍历(shape: (n_windows, window_size, n_cols))
for window_arr in sliding_window_view(df.values, window_shape=(window_size, len(cols))):
    # 构造临时 DataFrame(零拷贝,不复制原始数据)
    window_df = pd.DataFrame(window_arr[0], columns=cols, copy=False)

    # ✅ 自由按列名操作:可调用 .sum(), .mean(), .prod() 等,支持复杂逻辑
    D_val = window_df["A"].sum()                    # 标量
    E_val = (window_df["A"] + window_df["B"]).mean() # 标量
    F_val = (window_df["C"] - 1).prod()             # 标量
    G_val = (window_df["B"] * 2).sum()              # 标量

    results.append((D_val, E_val, F_val, G_val))

# 合并结果到原 DataFrame
result_df = pd.DataFrame(results, columns=["D", "E", "F", "G"])
df_final = pd.concat([df, result_df], axis=1)

print(df_final)

关键优势说明

Tome
Tome

先进的AI智能PPT制作工具

下载
  • 列名友好:window_df["A"] 直接访问,无需索引或位置硬编码
  • 输出自由:返回任意长度元组,映射为新列,不受输入列数约束;
  • 性能可控:sliding_window_view 是内存视图(非复制),copy=False 进一步避免冗余拷贝;
  • 可扩展性强:可在 window_df 上调用任意 Pandas 方法(groupby, agg, 自定义函数等)。

⚠️ 注意事项

  • sliding_window_view 返回的是 ndarray 视图,确保原始 df.values 不被修改,否则影响结果;
  • 若需跨行聚合(如窗口内每列独立统计),注意 window_arr[0] 提取的是第一个窗口切片(二维子数组),其行为与 df.iloc[i:i+window_size] 一致;
  • 对于超大规模数据(>千万行),可考虑分块处理或改用 numba.jit 加速核心计算逻辑。

该方法在代码清晰度、灵活性与执行效率之间取得了良好平衡,是替代低效循环或受限 rolling.apply 的生产级实践方案。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

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

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

68

2025.12.04

while的用法
while的用法

while的用法是“while 条件: 代码块”,条件是一个表达式,当条件为真时,执行代码块,然后再次判断条件是否为真,如果为真则继续执行代码块,直到条件为假为止。本专题为大家提供while相关的文章、下载、课程内容,供大家免费下载体验。

97

2023.09.25

硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

1155

2023.10.19

PHP接口编写教程
PHP接口编写教程

本专题整合了PHP接口编写教程,阅读专题下面的文章了解更多详细内容。

213

2025.10.17

php8.4实现接口限流的教程
php8.4实现接口限流的教程

PHP8.4本身不内置限流功能,需借助Redis(令牌桶)或Swoole(漏桶)实现;文件锁因I/O瓶颈、无跨机共享、秒级精度等缺陷不适用高并发场景。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

1901

2025.12.29

java接口相关教程
java接口相关教程

本专题整合了java接口相关内容,阅读专题下面的文章了解更多详细内容。

22

2026.01.19

go语言 数组和切片
go语言 数组和切片

本专题整合了go语言数组和切片的区别与含义,阅读专题下面的文章了解更多详细内容。

46

2025.09.03

go语言 数组和切片
go语言 数组和切片

本专题整合了go语言数组和切片的区别与含义,阅读专题下面的文章了解更多详细内容。

46

2025.09.03

C++ 设计模式与软件架构
C++ 设计模式与软件架构

本专题深入讲解 C++ 中的常见设计模式与架构优化,包括单例模式、工厂模式、观察者模式、策略模式、命令模式等,结合实际案例展示如何在 C++ 项目中应用这些模式提升代码可维护性与扩展性。通过案例分析,帮助开发者掌握 如何运用设计模式构建高质量的软件架构,提升系统的灵活性与可扩展性。

9

2026.01.30

热门下载

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

精品课程

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

共48课时 | 8.1万人学习

Excel 教程
Excel 教程

共162课时 | 14.4万人学习

PHP基础入门课程
PHP基础入门课程

共33课时 | 2万人学习

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

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