重新排序3D阵列的行 - python

我有一个像这样的值的3D数组:

[[[ 0  0  1  0 -1  1  1  0  0]
  [ 0  0 -1  0  1 -1 -1  0  0]
  [ 0  0  1 -2 -1  1  0  0  0]
  [ 0  0 -1  2  1 -1  0  0  0]]

 [[ 0  0  0  2  0  0  1  0  0]
  [ 0  0  0  0  0  0  0  0  0]
  [ 0  0  0 -2  0  0 -1  0  0]
  [ 0  0  0  0  0  0  0  0  0]]

 [[ 0  0  1  0 -1  1  1  0  0]
  [ 0  0 -1  2  1 -1  0  0  0]
  [ 0  0  1 -2 -1  1  0  0  0]
  [ 0  0 -1  0  1 -1 -1  0  0]]

 [[ 0  0  0  0  0  0  0  0  0]
  [ 0  0  0  0  0  0  0  0  0]
  [ 0  0  0  0  0  0  0  0  0]
  [ 0  0  0  0  0  0  0  0  0]]]

我需要重塑形状，以便每个矩阵中的所有第一行都被组合在一起，然后是所有第二行，依此类推。

这样的结果看起来像:

[[[ 0  0  1  0 -1  1  1  0  0]
  [ 0  0  0  2  0  0  1  0  0]
  [ 0  0  1  0 -1  1  1  0  0]
  [ 0  0  0  0  0  0  0  0  0]]

 [[ 0  0 -1  0  1 -1 -1  0  0]
  [ 0  0  0  0  0  0  0  0  0]
  [ 0  0 -1  2  1 -1  0  0  0]
  [ 0  0  0  0  0  0  0  0  0]]

 [[ 0  0  1 -2 -1  1  0  0  0]
  [ 0  0  0 -2  0  0 -1  0  0]
  [ 0  0  1 -2 -1  1  0  0  0]
  [ 0  0  0  0  0  0  0  0  0]]

 [[ 0  0 -1  2  1 -1  0  0  0]
  [ 0  0  0  0  0  0  0  0  0]
  [ 0  0 -1  0  1 -1 -1  0  0]
  [ 0  0  0  0  0  0  0  0  0]]]

python大神给出的解决方案

您只需要swap the first two axes数组x。如果数组是ndarray(与您的数组相同)，则返回视图，并且不复制任何数据:

>>> x.swapaxes(0,1)

例如:

>>> x = np.arange(27).reshape(3,3,3)
>>> x
array([[[ 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]]])

>>> x.swapaxes(0,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]]])