python numpyの配列の行と列を入れ替える

python numpyの配列の行と列を入れ替える

pythonで、ライブラリnumpyのtransposeを使用して、配列の行と列を入れ替えるサンプルコードを記述してます。pythonのバージョンは3.8.5を使用してます。

環境

  • OS windows10 pro 64bit
  • python 3.8.5

numpyインストール

numpyをインストールされていない方は、pipでインストールしておきます。

pip install numpy

# Successfully installed numpy-1.19.4

transpose使い方

transposeを使用すると、配列の行と列を入れ替えるを行うことが可能です。

import numpy as np

arr = np.array(配列) 

以下は、用意した配列の行と列を入れ替えるサンプルコードとなります。

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]]) 
print(np.transpose(arr))
# [[1 3 5]
#  [2 4 6]]

# 入れ替えたものをもとに戻す
arr = np.transpose(arr) 
print(np.transpose(arr))
# [[1 3 5]
#  [2 4 6]]