numpy配列の変形方法

Question

前提・実現したいこと

現在，chainerを使用した機械学習のプログラムを作成しています．学習を行うためのデータセット作りで行き詰ってます．

例

具体的なデータは貼ることができないので例を使って説明します．

np.reshape(np.arange(27),(3,3,3))

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]]])

この3次元のnumpy配列を

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]]])

に変形したいです．
僕にはfor文を使って値を入れ替える方法しか思いつかなかったので，もっとスマートな方法があれば教えていただけると嬉しいです．

Answer 1

np.transposeを使ってください。

>>> import numpy as np
>>> a = np.reshape(np.arange(27),(3,3,3))
>>> a
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]]])
>>> a.transpose((1,0,2))
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]]])

参考：
配列の軸の順番を入れ替えるNumPyのtranspose関数の使い方 - DeepAge

Comments

[pumper]

transpose自体は知っていましたがここで使えるとは．．
もっと参考URL読み込みます
ありがとうございます！

Answer 2

numpy.transpose() をお使いください

https://docs.scipy.org/doc/numpy-1.14.0/reference/generated/numpy.transpose.html

import numpy as np
a = np.reshape(np.arange(27),(3,3,3))

print(a.transpose(1,0,2))
#[[[ 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]]]

Comments

[pumper]

transpose自体は知っていましたがここで使えるとは．．
もっと参考URL読み込みます
ありがとうございます！