раздвижное окно в NumPy
У меня есть бесформенный массив формы (6,2)
[[00,01],
[10,11],
[20,21],
[30,31],
[40,41],
[50,51]]
Мне нужно раздвижное окно с размером шага 1 и размером окна 3 нравится это:
[[00,01,10,11,20,21],
[10,11,20,21,30,31],
[20,21,30,31,40,41],
[30,31,40,41,50,51]]
Я ищу тупое решение. Если бы ваше решение могло параметризовать форму исходного массива, а также размер окна и размер шага, это было бы здорово.
Я нашел этот связанный ответИспользование шагов для эффективного фильтра скользящих средних но я не вижу, как указать размер шага там и как свернуть окно из 3d в непрерывный 2d массив. Также этоИтератор с подвижным или скользящим окном в Python но это в Python, и я не уверен, насколько это эффективно. Кроме того, он поддерживает элементы, но не объединяет их в конце концов, если каждый элемент имеет несколько функций.