スポンサーリンク

[numpy] 単位行列の2次元配列作成(Python)

単位行列の2次元配列作成

np.eye関数を使うことで単位行列を作成できます。引数にはサイズを指定します。

IN

import numpy as np
np.eye(3)

OUT

array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])