python - 用于将 ndarray 拆分为更小的重叠 ndarray 的滑动窗口

Question

我目前正在研究一种可以拆分 numpy.ndarray 的方法进入给定数量的子阵列，只要该数量小于窗口移动的轴。

示例:

给定一个形状为 (15, 40, 3) 的 numpy.ndarray 我想分成 36 个大小相等的子数组。重叠是允许的，因为在大多数情况下数组的长度和子数组的数量不能被整除。

每个子数组的shape必须是(15, ?, 3)，其中?是滑动窗口移动的轴。允许重叠。

我该怎么做？ numpy.splits 要求项目是可等分的，也不允许重叠。

预期输出:给定一个形状为 (5,3) 的 numpy.ndarray，我想提取 2 个形状为 (5,2) 的 numpy.ndarray，滑动窗口移动轴 = 1。它将始终是这个轴。

我不想明确地想要设置两个 numpy.ndarrays 之间应该有多少重叠，但只有应该提取的 ndarrays 的数量..

所以在这种情况下，每个子 numpy.ndarray 应该这样放置:

Answer 1

可以在这里找到似乎是什么的实现:

https://popi.ucdavis.edu/~travc/tmp/dev/racc/sliding_window.py