janela deslizante em numpy
Eu tenho uma matriz numpy de forma (6,2)
[[00,01],
[10,11],
[20,21],
[30,31],
[40,41],
[50,51]]
Eu preciso de uma janela deslizante com tamanho de passo 1 e tamanho de janela 3 gosta disso:
[[00,01,10,11,20,21],
[10,11,20,21,30,31],
[20,21,30,31,40,41],
[30,31,40,41,50,51]]
Eu estou procurando uma solução numpy. Se a sua solução pudesse parametrizar a forma da matriz original, bem como o tamanho da janela e o tamanho da etapa, seria ótimo.
Eu encontrei esta resposta relacionadaUsando passadas para um filtro eficiente de média móvel mas não vejo como especificar o tamanho das etapas e como recolher a janela do 3d para um array 2d contínuo. Também issoRolling or sliding iterator window em Python mas isso é em Python e não tenho certeza de como isso é eficiente. Além disso, ele suporta elementos, mas não os une no final, se cada elemento tiver vários recursos.