3D np配列の各要素を転置する方法

Question

3D配列aを指定すると、最初のインデックスの各要素に対してnp.tranposeを呼び出す必要があります。例えば
は、アレイ所与：

array([[[1, 1, 1, 1], 
     [1, 1, 1, 1], 
     [1, 1, 1, 1]], 

     [[2, 2, 2, 2], 
     [2, 2, 2, 2], 
     [2, 2, 2, 2]], 

     [[3, 3, 3, 3], 
     [3, 3, 3, 3], 
     [3, 3, 3, 3]]) 

欲しい：

array([[[1, 1, 1], 
     [1, 1, 1], 
     [1, 1, 1], 
     [1, 1, 1]], 

     [[2, 2, 2], 
     [2, 2, 2], 
     [2, 2, 2], 
     [2, 2, 2]], 

     [[3, 3, 3], 
     [3, 3, 3], 
     [3, 3, 3], 
     [3, 3, 3]]]) 

は、本質的に、私は、配列内の各要素を転置します。私はそれを形作ろうとしましたが、それを行う良い方法を見つけることができません。それをループし、それぞれを転置するのは遅すぎるでしょう。何かアドバイス？

Answer 1

あなたは、組み込みのnumpyのtransposeメソッドを使用して直接

>>> a = np.array([[[1, 1, 1, 1], [1, 1, 1, 1], [1, 1, 1, 1]], 
        [[2, 2, 2, 2], [2, 2, 2, 2], [2, 2, 2, 2]], 
        [[3, 3, 3, 3], [3, 3, 3, 3], [3, 3, 3, 3]]]) 

>>> print(a.transpose((0, 2, 1))) 

[[[1 1 1] 
    [1 1 1] 
    [1 1 1] 
    [1 1 1]] 

[[2 2 2] 
    [2 2 2] 
    [2 2 2] 
    [2 2 2]] 

[[3 3 3] 
    [3 3 3] 
    [3 3 3] 
    [3 3 3]]]