圆形阵列上的滚动窗口

Question

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

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

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

Answer 1

填充足够的值，使“伪圆形”数组的原始数组。然后应用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]]