讲师中心 微信公众号

工具推荐

视频效率加速
AniShort
AniShort

AI短剧协同制作平台

Seko
Seko

AI短剧一键生成成片

updream

一键生成高质量推广成片

Laper
Laper

剧本策划到成片协作

Pandas 滚动窗口自定义函数应用:支持列名访问与任意输出维度的高效方案

千辰君_8573

千辰君_8573

发布时间:2026-01-30 20:04:02

|

223人浏览过

|

来源于php中文网

原创

Pandas 滚动窗口自定义函数应用:支持列名访问与任意输出维度的高效方案

本文介绍如何在 pandas 中实现真正灵活的滚动窗口计算——无需显式循环、保留 dataframe 列名访问能力,并支持输出任意数量/名称的新列。核心方案结合 `numpy.sliding_window_view` 与轻量级临时 dataframe 构建,兼顾可读性、表达力与性能。

Pandas 原生的 .rolling().apply() 在处理复杂窗口逻辑时存在明显限制:启用 raw=True 后函数接收的是 numpy.ndarray,无法通过列名(如 window["A"])访问数据;而关闭 raw 时又强制要求返回值形状与输入列数一致,难以生成多于或少于原列数的新特征。

一个简洁、可靠且生产就绪的替代方案是使用 NumPy 的 sliding_window_view ——它能以零拷贝方式生成滑动窗口视图,再配合极轻量的 pd.DataFrame(..., copy=False) 构造临时子表。这样既保持了列名语义(df_tmp["A"].mean()),又完全解耦输入列数与输出结构。

以下为完整可运行示例:

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

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

# 构造示例数据
df = pd.DataFrame({
    "A": range(10),
    "B": range(10, 20),
    "C": range(20, 30)
})

# 定义窗口大小(行数 × 列数),此处为 2 行 × 全部 3 列
window_rows, window_cols = 2, df.shape[1]
cols = df.columns.tolist()

# 预分配结果列表,首行为 NaN 占位(因窗口大小为 2,第 0 行无前置窗口)
results = [tuple([np.nan] * 4)]  # 对应新列 D, E, F, G

# 遍历每个滑动窗口(形状为 (2, 3) 的二维数组)
for window_arr in sliding_window_view(df.values, window_shape=(window_rows, window_cols)):
    # 构建无拷贝临时 DataFrame,保留原始列名
    df_tmp = pd.DataFrame(window_arr[0], columns=cols, copy=False)  # window_arr[0] 取首窗口块(2×3)

    # ✅ 自由使用列名进行任意计算(支持向量化操作)
    D = df_tmp["A"].sum()                    # 标量:A 列窗口和
    E = (df_tmp["A"] + df_tmp["B"]).mean()   # 标量:A+B 的窗口均值
    F = (df_tmp["C"] - 1).prod()             # 标量:C-1 的窗口乘积
    G = (df_tmp["B"] * 2).sum()              # 标量:B×2 的窗口和

    results.append((D, E, F, G))

# 批量拼接新列
new_cols_df = pd.DataFrame(results, columns=["D", "E", "F", "G"])
df_result = pd.concat([df, new_cols_df], axis=1)

print(df_result)

✅ 关键优势说明:

  • 列名友好:df_tmp["A"] 直观可读,支持 .groupby, .agg, 自定义函数等全部 DataFrame 语法;
  • 输出自由:返回任意长度元组 → 自动映射为指定列名,不受原始列数约束;
  • 内存高效:sliding_window_view 返回视图(view),copy=False 确保临时 DataFrame 不复制底层数据;
  • 可扩展性强:可轻松嵌入 def compute_features(df_slice): ... 封装复杂逻辑,便于单元测试与复用。

⚠️ 注意事项:

  • sliding_window_view 要求 NumPy ≥ 1.20;旧版本可用 np.lib.stride_tricks.as_strided 手动实现,但需谨慎处理内存边界;
  • 窗口对齐默认为右对齐(即窗口覆盖 [i-1, i] 得到第 i 行结果),若需左对齐或居中,可调整 results 初始化与索引偏移;
  • 对超大规模数据(千万行+),可考虑分块处理或转向 Dask/Polars,但本方案在百万级数据上通常优于纯 Python 循环 10–50 倍。

该方法在保持代码清晰性的同时,突破了 .rolling().apply() 的固有瓶颈,是构建时间序列特征工程、滚动统计指标或模型滑动预测 pipeline 的推荐实践。

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

相关专题

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

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

360

2025.12.04

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

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

272

2026.01.31

Python数据分析与Pandas高级实战
Python数据分析与Pandas高级实战

本专题围绕 Python 数据分析展开,系统讲解 Pandas 的高级用法,包括数据清洗、透视表、时间序列分析以及多表合并与分组操作。通过实战案例,帮助开发者掌握高效处理与分析数据的方法,提高数据处理效率与分析能力。

891

2026.04.13

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

60

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

40

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

40

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

60

2026.11.05

DBeaver安全防护教程大全
DBeaver安全防护教程大全

DBeaver安全防护专题,围绕数据库连接安全、账号密码保护和访问权限管理展开,整理SSL/TLS加密连接、SSH隧道、数据库凭据保存及证书验证等配置方法,同时介绍敏感信息保护、远程连接风险和高权限账号使用等注意事项,帮助用户提高数据库日常管理的安全性。

40

2026.11.05

DBeaver连接创建与配置教程
DBeaver连接创建与配置教程

本专题聚焦DBeaver数据库连接配置与管理,覆盖从新建连接、选择数据库类型到填写连接参数和验证连接的完整流程。针对MySQL、PostgreSQL、SQL Server等常见数据库,整理相应的连接设置方法,并提供驱动缺失、认证失败和连接异常等问题的排查方向。

60

2026.11.04

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn