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

Zeineb

我有一个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]]]
迪卡卡

只需置换轴-

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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章