numpy transpose

import numpy as np

arr = np.arange(108).reshape((6, 6, 3))
print('arr:', arr)
print('arr shape:', arr.shape)

arr_tras = arr.transpose((2, 0, 1))
print('arr_tras', arr_tras)
print('arr_tras shape', arr_tras.shape)
arr: [[[  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]
  [ 27  28  29]
  [ 30  31  32]
  [ 33  34  35]]

 [[ 36  37  38]
  [ 39  40  41]
  [ 42  43  44]
  [ 45  46  47]
  [ 48  49  50]
  [ 51  52  53]]

 [[ 54  55  56]
  [ 57  58  59]
  [ 60  61  62]
  [ 63  64  65]
  [ 66  67  68]
  [ 69  70  71]]

 [[ 72  73  74]
  [ 75  76  77]
  [ 78  79  80]
  [ 81  82  83]
  [ 84  85  86]
  [ 87  88  89]]

 [[ 90  91  92]
  [ 93  94  95]
  [ 96  97  98]
  [ 99 100 101]
  [102 103 104]
  [105 106 107]]]
arr shape: (6, 6, 3)
arr_tras [[[  0   3   6   9  12  15]
  [ 18  21  24  27  30  33]
  [ 36  39  42  45  48  51]
  [ 54  57  60  63  66  69]
  [ 72  75  78  81  84  87]
  [ 90  93  96  99 102 105]]

 [[  1   4   7  10  13  16]
  [ 19  22  25  28  31  34]
  [ 37  40  43  46  49  52]
  [ 55  58  61  64  67  70]
  [ 73  76  79  82  85  88]
  [ 91  94  97 100 103 106]]

 [[  2   5   8  11  14  17]
  [ 20  23  26  29  32  35]
  [ 38  41  44  47  50  53]
  [ 56  59  62  65  68  71]
  [ 74  77  80  83  86  89]
  [ 92  95  98 101 104 107]]]
arr_tras shape (3, 6, 6)

猜你喜欢

转载自blog.csdn.net/weixin_40437821/article/details/113760902
今日推荐