Question

我想将一个给定的函数（特别是np.std（））应用于一个测量数组，我想将它应用于给定大小的滚动窗口。

但是 - 由于测量是在圆形阵列中 - 我还需要滚动窗口能够从阵列端到其开头重叠。

因此，我无法在Rolling window for 1D arrays in Numpy?中使用答案...我试图修改它的方法，但我不是一个笨拙的专家，我无法理解np.lib.stride_tricks是什么。 as_strided做（其文档在哪里？）

Answer 1

用足够的值填充原始数组以制作＆＃34;伪循环＆＃34;阵列。然后将rolling_window应用于伪圆形数组：

import numpy as np

def rolling_window(a, window):
    # http://www.mail-archive.com/numpy-discussion@scipy.org/msg29450.html
    shape = a.shape[:-1] + (a.shape[-1] - window + 1, window)
    strides = a.strides + (a.strides[-1],)
    return np.lib.stride_tricks.as_strided(a, shape=shape, strides=strides)

def circular_rolling_window(a, window):
    pseudocircular = np.pad(a, pad_width=(0, windowsize-1), mode='wrap')
    return rolling_window(pseudocircular, windowsize)

a = np.arange(5)
windowsize = 3
print(circular_rolling_window(a, windowsize))

产量

[[0 1 2]
 [1 2 3]
 [2 3 4]
 [3 4 0]
 [4 0 1]]

圆形阵列上的滚动窗口

1 个答案: