Pythonで行列を1行で転置する方法！NumPyのtranspose()の使い方

Pythonで行列（2次元配列）を転置したいとき、最も手軽な方法はNumPyパッケージに含まれるtranspose()関数を使うことです。ndarrayオブジェクトに対してこのメソッドを呼び出すだけで、行と列が入れ替わった転置配列を1行のコードで取得できます。

NumPyのtranspose()で転置する基本例

>>> import numpy as np
>>> a = np.array([[1, 2], [3, 4]])
>>> a
array([[1, 2],
       [3, 4]])
>>> a.transpose()
array([[1, 3],
       [2, 4]])

上の例では、元の配列 [[1, 2], [3, 4]] の行と列が入れ替わり、[[1, 3], [2, 4]] という転置行列が得られています。

.T属性でさらに簡潔に書く

NumPyには transpose() のショートカットとして .T属性 も用意されています。コードが短くなり可読性も向上するため、実際の開発ではこちらがよく使われます。

>>> a.T
array([[1, 3],
       [2, 4]])

NumPyを使わずリストのまま転置する方法

通常のリストの状態で転置したい場合は、組み込み関数 zip(*matrix) を利用できます。結果はタプルのリストになるため、必要に応じて list(map(list, ...)) でリストに変換しましょう。

>>> matrix = [[1, 2], [3, 4]]
>>> list(zip(*matrix))
[(1, 3), (2, 4)]

まとめ

• NumPyのndarrayなら transpose() メソッドまたは .T 属性で1行で転置できる
• リストのまま処理したい場合は zip(*matrix) が便利
• 大規模な数値データを扱うなら、高速に動作するNumPyの利用がおすすめ