python - 堆叠矩阵的最佳方法

Question

我有两个矩阵

[[0,0,0],
 [0,0,0],
 [0,0,0]]

[[1,1,1],
 [1,1,1],
 [1,1,1]]

我想要它们像这样 v-stacked 和 h-stacked

[[0,0,0,1,1,1],
 [0,0,0,1,1,1],
 [0,0,0,1,1,1],
 [1,1,1,0,0,0],
 [1,1,1,0,0,0],
 [1,1,1,0,0,0]]

做这个的最好方式是什么？

Answer 1

你可以用 np.block为此：

A = np.array([[0,0,0],
              [0,0,0],
              [0,0,0]])

B = np.array([[1,1,1],
              [1,1,1],
              [1,1,1]])

np.block([[A, B], [B, A]])

输出：

array([[0, 0, 0, 1, 1, 1],
       [0, 0, 0, 1, 1, 1],
       [0, 0, 0, 1, 1, 1],
       [1, 1, 1, 0, 0, 0],
       [1, 1, 1, 0, 0, 0],
       [1, 1, 1, 0, 0, 0]])