【Python応用】PythonとNumPy-eyeを用いた単位行列の作成方法-

ここではNumPyにおけるeyeを用いた単位行列を作成する方法について解説しています。

基本的な型

numpy.eye(N,dtype=<class float>)

基本的な単位行列の生成する方法は、Nに出力の行数、dtypeにデータ型を指定することで可能です。

import numpy as np
np_int = np.eye(5, dtype=int)
print(np_int)

引数に5(5次の単位行列)、intを指定すると結果は以下のようになります。

[[1 0 0 0 0]
 [0 1 0 0 0]
 [0 0 1 0 0]
 [0 0 0 1 0]
 [0 0 0 0 1]]

単位行列を生成する方法には「identity」を用いる方法もあります。

import numpy as np
np_int = np.identity(5)
print(np_int)

使用方法としてはeyeと一緒で引数に数値とデータ型を指定するだけです。ここではデータ型はデフォルトのままなのでfloat型として出力されています。以下が、結果になります。

[[1. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]]