0

0

numpy 如何高效实现滑动窗口计算(不使用 for 循环)

舞夢輝影

舞夢輝影

发布时间:2026-01-24 14:21:09

|

614人浏览过

|

来源于php中文网

原创

NumPy 1.20+ 官方推荐 sliding_window_view,基于内存视图零拷贝、速度快;一维输入得二维输出,聚合需指定 axis=1;支持多维但步长固定为1,边界不补零仅保留完整窗口。

numpy 如何高效实现滑动窗口计算(不使用 for 循环)

numpy.lib.stride_tricks.sliding_window_view 最直接

这是 NumPy 1.20+ 官方推荐的无循环滑动窗口方案,底层基于内存视图,不复制数据,速度极快且语义清晰。

常见错误是传入窗口大小后忽略输出维度变化:对一维数组 a 调用 sliding_window_view(a, window_shape=3) 得到的是 shape 为 (len(a)-2, 3) 的二维数组,后续聚合需明确轴向(通常是 axis=1)。

  • 支持多维:如对图像(H,W)做 3×3 窗口,sliding_window_view(img, (3,3)) 输出 shape 为 (H-2, W-2, 3, 3)
  • 不支持步长 ≠ 1:若需跨步(如 stride=2),得先切片再调用,例如 sliding_window_view(a[::2], 3)
  • 边界不补零也不截断:默认只生成完整窗口,长度不足的末尾直接丢弃

手动构造索引 + np.take 或高级索引(兼容旧版 NumPy)

当环境受限(如 NumPy 内存占用略高(会临时存所有窗口数据)。

关键点在于索引数组的构造:对长度为 N 的数组,窗口长 L,则起始位置为 np.arange(N - L + 1);再叠加 np.arange(L) 构成偏移矩阵,最后用 arr[indices] 一次性取出所有窗口。

  • 示例(一维):idx = np.arange(len(a) - 2)[:, None] + np.arange(3); windows = a[idx]
  • np.take 在某些场景下比高级索引更稳(尤其当 a 是非 C 连续数组时)
  • 注意 idx 形状必须是二维,否则广播失败;[:, None] 是关键,别写成 [..., None]

避免误用 np.convolve 做通用窗口计算

np.convolve 本质是相关运算,仅适用于线性加权求和(如移动平均、卷积核),不能直接用于中位数、最大值等非线性操作。

AItools.fyi
AItools.fyi

找到让生活变得更轻松的最佳AI工具!

下载

常见误用:想算滑动中位数却套用 convolve,结果完全错误。它内部做的是 sum(window * kernel),kernel 必须提前给定且长度固定。

  • 适合场景:滑动平均(kernel = np.ones(w)/w)、梯度近似([-1, 0, 1]
  • 边界模式有限:mode='valid'(默认)丢弃边界,'same' 补零,无法实现镜像填充或自定义 padding
  • 性能虽好,但功能窄——不是万能替代品

性能与内存的关键取舍点

所有无循环方法都在「时间换空间」或「空间换时间」之间权衡,实际选型要看数据规模和后续操作。

  • sliding_window_view 内存最优(零拷贝),但返回视图对象,若后续做大量写入或 reshape,可能触发隐式复制
  • 手动索引法(arange + broadcasting)会分配完整窗口数据内存,例如 1e6 长度、窗口长 100 → 1e6×100 元素,约 800MB float64,极易爆内存
  • 大数据建议分块处理:用 sliding_window_view 切出小段,逐段聚合(如 windows.mean(axis=1)),避免中间数组膨胀

最易被忽略的是:窗口聚合后是否还需保留窗口结构?如果只要一个标量序列(如每窗一个均值),就别留着二维数组——立刻降维,否则后续操作成本指数上升。

相关专题

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

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

46

2025.09.03

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

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

46

2025.09.03

css中的padding属性作用
css中的padding属性作用

在CSS中,padding属性用于设置元素的内边距。想了解更多padding的相关内容,可以阅读本专题下面的文章。

133

2023.12.07

windows查看端口占用情况
windows查看端口占用情况

Windows端口可以认为是计算机与外界通讯交流的出入口。逻辑意义上的端口一般是指TCP/IP协议中的端口,端口号的范围从0到65535,比如用于浏览网页服务的80端口,用于FTP服务的21端口等等。怎么查看windows端口占用情况呢?php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

688

2023.07.26

查看端口占用情况windows
查看端口占用情况windows

端口占用是指与端口关联的软件占用端口而使得其他应用程序无法使用这些端口,端口占用问题是计算机系统编程领域的一个常见问题,端口占用的根本原因可能是操作系统的一些错误,服务器也可能会出现端口占用问题。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1126

2023.07.27

windows照片无法显示
windows照片无法显示

当我们尝试打开一张图片时,可能会出现一个错误提示,提示说"Windows照片查看器无法显示此图片,因为计算机上的可用内存不足",本专题为大家提供windows照片无法显示相关的文章,帮助大家解决该问题。

799

2023.08.01

windows查看端口被占用的情况
windows查看端口被占用的情况

windows查看端口被占用的情况的方法:1、使用Windows自带的资源监视器;2、使用命令提示符查看端口信息;3、使用任务管理器查看占用端口的进程。本专题为大家提供windows查看端口被占用的情况的相关的文章、下载、课程内容,供大家免费下载体验。

454

2023.08.02

windows无法访问共享电脑
windows无法访问共享电脑

在现代社会中,共享电脑是办公室和家庭的重要组成部分。然而,有时我们可能会遇到Windows无法访问共享电脑的问题。这个问题可能会导致数据无法共享,影响工作和生活的正常进行。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2351

2023.08.08

c++ 根号
c++ 根号

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

25

2026.01.23

热门下载

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

精品课程

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

共48课时 | 7.7万人学习

Excel 教程
Excel 教程

共162课时 | 13.2万人学习

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

共33课时 | 2万人学习

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

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