【NumPy】単位行列の配列生成 (numpy.eye)

この記事では、Python言語とNumPyを用いて、単位行列の配列を作成する方法をソースコード付きで解説します。

単位行列の配列生成

Pythonの数値計算モジュールNumPyでは、numpy.eye(m)メソッドでm×mの単位行列(配列)を作成できます。

ソースコード

サンプルプログラムのソースコードは下記の通りです。

# -*- coding: utf-8 -*-
import numpy as np

def main():
    # 配列の宣言・初期化
    X = np.eye(3)
    # 画面出力
    print(X)

if __name__ == '__main__':
    main()

実行結果

サンプルプログラムの実行結果です。

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

【おすすめ関連記事】
Python入門 基本文法
NumPy入門 サンプルプログラム集

シェア&フォローお願いします!