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