首页 > 后端开发 > Python教程 > 如何在 NumPy 中高效计算滚动标准差?

如何在 NumPy 中高效计算滚动标准差?

Linda Hamilton
发布: 2024-10-27 16:17:02
原创
458 人浏览过

How to Efficiently Calculate Rolling Standard Deviations in NumPy?

在 NumPy 中实现一维数组的滚动窗口

在数据分析和操作中经常出现对一维数组高效执行滚动窗口操作的必要性。 NumPy 是一个强大的数值运算库,为此任务提供了多功能解决方案。

例如,考虑计算一维值列表的滚动标准差的任务。正如问题中给出的,使用循环的基本 Python 实现对于大型数据集来说计算成本可能很高。

基于 NumPy 的解决方案

为了优化此过程,NumPy 提供了简洁高效的方法:

<code class="python">import numpy as np

def rolling_std(array, window):
    return np.std(np.lib.stride_tricks.as_strided(array, shape=array.shape[0] - window + 1, strides=(array.strides[0], array.strides[0])), 1)</code>
登录后复制

rolling_std 函数有两个参数:一维数组和所需的窗口大小窗口。

NumPy 的 as_strided 函数创建数组的视图,就好像它一样被划分为重叠的窗口,其步幅允许高效计算。

np.std 函数然后计算沿第二维的标准差,有效地执行滚动操作。

用法

要计算输入数组观测值的滚动标准差,只需调用:

<code class="python">stdev = rolling_std(observations, window_size)</code>
登录后复制

此解决方案具有高性能,无需显式循环,并充分利用NumPy 的高效数组操作功能。

以上是如何在 NumPy 中高效计算滚动标准差?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板