提问人：Massimo 提问时间：1/9/2023 更新时间：1/9/2023 访问量：133
麻木。如何通过网格将二维数组拆分为多个数组？
Numpy. How to split 2D array to multiple arrays by grid?
问：
我有一个numpy 2D数组：
c = np.arange(36).reshape(6, 6)
[[ 0, 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10, 11],
[12, 13, 14, 15, 16, 17],
[18, 19, 20, 21, 22, 23],
[24, 25, 26, 27, 28, 29],
[30, 31, 32, 33, 34, 35]]
我想通过网格 3x3 将其拆分为多个 2D 数组。（这就像通过网格 3x3 将大图像拆分为 9 个小图像）：
[[ 0, 1,| 2, 3,| 4, 5],
[ 6, 7,| 8, 9,| 10, 11],
---------+--------+---------
[12, 13,| 14, 15,| 16, 17],
[18, 19,| 20, 21,| 22, 23],
---------+--------+---------
[24, 25,| 26, 27,| 28, 29],
[30, 31,| 32, 33,| 34, 35]]
最后，我需要具有 9 个 2D 数组的数组。喜欢这个：
[[[0, 1], [6, 7]],
[[2, 3], [8, 9]],
[[4, 5], [10, 11]],
[[12, 13], [18, 19]],
[[14, 15], [20, 21]],
[[16, 17], [22, 23]],
[[24, 25], [30, 31]],
[[26, 27], [32, 33]],
[[28, 29], [34, 35]]]
这只是我需要的一个样本。我想知道如何通过网格 （N，M） 从大的 2D 数组制作小型 2D 数组
答：
2赞
Corralien
1/9/2023
#1
您可以使用类似以下内容的内容：
from numpy.lib.stride_tricks import sliding_window_view
out = np.vstack(sliding_window_view(c, (2, 2))[::2, ::2])
输出：
>>> out.tolist()
[[[0, 1], [6, 7]],
[[2, 3], [8, 9]],
[[4, 5], [10, 11]],
[[12, 13], [18, 19]],
[[14, 15], [20, 21]],
[[16, 17], [22, 23]],
[[24, 25], [30, 31]],
[[26, 27], [32, 33]],
[[28, 29], [34, 35]]]
评论
0赞
Massimo
1/9/2023
TNX的。这正是我所需要的！
评论