如何重新排列3D Numpy数组?

时间:2019-10-29 10:27:55

标签: python numpy numpy-ndarray

我有一个numpy ndarray,像这样(我的数组更大,我只是为了解释我需要做的事情而给它):

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

我正在寻找一种优雅的方式来重新排列数组,以获得类似这样的内容:

a_new=[[[ 0   9  18]
        [ 1  10  19]
        [ 2  11  20]]

       [[ 3 12 21]
        [ 4 13 22]
        [ 5 14 23]]

       [[ 6 15 24]
        [ 7 16 25]
        [ 8 17 26]]]

1 个答案:

答案 0 :(得分:3)

仅置换轴-

a.transpose(1,2,0)
# or np.moveaxis(a,0,2)