python - 将 Numpy 方阵数组转换为 numpy 方阵，保留 "layout"

Question

第一次在这里问问题。我有点“受阻”。

我有一个由 n x n 数组组成的数组(为简化起见，让 n=3):

[
    [
        [ 0,  1,  2],
        [ 3,  4,  5],
        [ 6,  7,  8]
    ],
   [
        [9, 10, 11],
        [12, 13, 14],
        [15, 16, 17]
    ],
   [
        [18, 19, 20],
        [21, 22, 23],
        [24, 25, 26]
    ]
        ]

(尽管我的数组包含的 3*3 数组多于 3 个)

我想实现这样的二维数组:

[0,1,2,9,10,11,18,19,20]  
[3,4,5,12,13,14,21,22,23]    
[6,7,8,15,16,17,24,25,26]

有没有想不到的技巧，因为我想不出任何方法来完成转换。

谢谢

Answer 1

可能比 moveaxis 稍微干净一些:

import numpy as np
a = np.arange(27).reshape(3,3,3)
a.swapaxes(0,1).reshape(3,-1)
array([[ 0,  1,  2,  9, 10, 11, 18, 19, 20],
   [ 3,  4,  5, 12, 13, 14, 21, 22, 23],
   [ 6,  7,  8, 15, 16, 17, 24, 25, 26]])