python-堆叠矩阵的最佳方法

我有两个矩阵

[[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]]

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

回答

你可以用 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]])


以上是python-堆叠矩阵的最佳方法的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>