在numpy中,滚动窗口是一种常用的数据处理技术,可以对数组进行移动窗口操作并应用特定的函数。下面是如何将函数应用于numpy中的滚动窗口的步骤:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
numpy.lib.stride_tricks.sliding_window_view
函数创建滚动窗口视图。该函数返回一个对原始数组进行滚动窗口处理后的视图数组。具体代码如下:from numpy.lib.stride_tricks import sliding_window_view
window_size = 3
window_view = sliding_window_view(arr, window_size)
def sum_window(window):
return np.sum(window)
numpy.apply_along_axis
函数应用定义的函数到滚动窗口视图的每个窗口中,以获得每个窗口的处理结果。具体代码如下:result = np.apply_along_axis(sum_window, -1, window_view)
在这个例子中,-1
表示沿着最后一个轴应用函数。result
将包含滚动窗口中应用函数后的结果。
总结起来,将函数应用于numpy中的滚动窗口的步骤如下:
numpy.lib.stride_tricks.sliding_window_view
函数创建滚动窗口视图。numpy.apply_along_axis
函数应用定义的函数到滚动窗口视图的每个窗口中,以获得每个窗口的处理结果。对于腾讯云相关产品,暂时无法提供相关推荐链接。
领取专属 10元无门槛券
手把手带您无忧上云