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.