在保留阵列位置的同时，将4d numpy阵列重塑为2d阵列

Question

{numpy}的多维数组。所以给定一个行和列索引(i, j)，mat[i,j]就是QxQ矩阵。我想把这个数组改成(N*Q, N*Q)，这样

array([[[[ 0,  1],
         [ 2,  3]],

        [[ 4,  5],
         [ 6,  7]]],


       [[[ 8,  9],
         [10, 11]],

        [[12, 13],
         [14, 15]]]])

去

^{pr2}$

您可以看到mat[0,0]指向new_mat[0:2, 0:2]。当前mat.reshape(N*Q, N*Q)将mat[0,0]带到{}（这是我不想要的）。如何使用“重塑”或“rollaxis”或类似的方法重塑此数组？我最终想用imshow来绘制它，现在卡住了。我觉得这很容易，只是还没弄明白。在

Answer 1

没关系，我想出来了。np.swapaxes(1, 2)是我所需要的缺失部分。在

答案就是做mat.swapaxes(1, 2).reshape(N*Q, N*Q)。在

我觉得自己太长时间不去想办法就发帖子是愚蠢的，但我会把它留着，这样别人就可以从中受益。在