【Python】Numpyのtransposeメソッドの文法や多次元配列の引数の使い方についてのまとめ

taransposeメソッドの公式ドキュメント

公式ドキュメントはこちら。
>>numpy.transpose — NumPy v1.15 Manual

numpy.transpose(a, axes=None)
Permute the dimensions of an array.

Parameters:

a : array_like
Input array.

axes : list of ints, optional
By default, reverse the dimensions, otherwise permute the axes
according to the values given.

 

Returns:

p : ndarray
a with its axes permuted. A view is returned whenever possible.

と書かれています。

transposeの使用例

import numpy as np 
a = np.arange(6) #長さが６の配列を作る 
a
#　=>　array([0, 1, 2, 3, 4, 5])

reshapeメソッドで２行３列の配列に変換

a = a.reshape(2, 3) 
a 
# => array([[0, 1, 2], 
# [3, 4, 5]])

引数なしでtranspose

a.transpose() 
# =>array([[0, 3], 
# [1, 4], 
# [2, 5]])

引数(0, 1)でtranspose

a.transpose(0, 1) 
#array([[0, 1, 2], 
# [3, 4, 5]])

引数(0, 1)でtranspose

a.transpose(1, 0) 
# => array([[0, 3], 
# [1, 4], 
# [2, 5]])

結構問題になってくるのがこの「引数(0, 1)でtranspose」と「引数(0, 1)でtranspose」の違いです。ドキュメントみてもよくわからない人が多いと思います。
(0, 1)はそのままで、(1, 0)は変換するんだ、などと暗記のように覚えている方も多いのではないでしょうか。

配列と同じように(0, 1, 2)という順番で次元を整理する

これは配列と同じように次元の順番ができている。という風に覚えました。

例えばこれだと2×3の行列式ですが、配列の0番目に0が、配列の1番目に1が入っている状態です。
どんな配列も自分で設定せずにデフォルトで(0, 1 )という設定がなされています。

３次元配列でも同じことが言える

３次元配列でも同じようなことが言えます。

(3, 2, 2)の１２個の行列を作成

b = np.arange(12) 
b = b.reshape(3, 2, 2) 
# array([[[ 0, 1], # [ 2, 3]], 
# [[ 4, 5], 
# [ 6, 7]], 
# [[ 8, 9], 
# [10, 11]]])

transposeを試してみる

b.transpose() 
# array([[[ 0, 4, 8], 
# [ 2, 6, 10]], 
# [[ 1, 5, 9], 
# [ 3, 7, 11]]])

これは(2, 1, 0)と同義

b.transpose(2, 1, 0) 
# array([[[ 0, 4, 8], 
# [ 2, 6, 10]], 
# [[ 1, 5, 9], 
# [ 3, 7, 11]]])

違った変換の仕方を試してみる

b.transpose(1, 2, 0)
# array([[[ 0,  4,  8],
#         [ 1,  5,  9]],
#        [[ 2,  6, 10],
#         [ 3,  7, 11]]])

以上です。