通过特定大小的迷你矩阵展平矩阵

Question

我想在 python 中展平一个 M×N 矩阵。

H = [[1,3,0,1],[6,5,4,8],[1,1,1,1],[3,2,0,0]]

为此，我需要选择 s x t 的窗口大小并首先展平该窗口，然后继续执行该过程，直到我的窗口覆盖所有 M x N 矩阵。

假设我的 s 和 t 是 2。 所以，我想得到：

H_flat = [1,3,6,5,0,1,4,8,1,1,3,2,1,1,0,0]

有没有一种快速的方法可以只使用循环或不使用循环？

Answer 1

试试：

res = []
for i in range(0, len(m), 2):
    res.extend(m[i][:2] + m[i + 1][:2] + m[i][2:] + m[i +1][2:])
print(res)
# [1, 3, 6, 5, 0, 1, 4, 8, 1, 1, 3, 2, 1, 1, 0, 0]