numpy.eye()関数

参考記事:https://docs.scipy.org/doc/numpy/reference/generated/numpy.eye.html numpy.eye(N,M=なし,k=0,dtype=,order=‘C’)は、対角線が1,零点がゼロの2次元配列を返します.
パラメータ:N:int出力の行数.M:int、出力中の列数を選択可能にします.もしなかったら,デフォルトは烩とする.k:int、オプション対角線のインデックス:0(デフォルト)はメイン対角線、正の値は上対角線、負の値は下対角線を指します.dtype:データ型、配列のデータ型を返すオプション.order:{‘C’,‘F’},オプション出力はメモリ内の行主(Cスタイル)または列主(Fortranスタイル)の順に格納すべきかどうか.
戻ります.すべての要素がゼロに等しく、k番目の対角線を除いて1に等しい配列です.

import numpy as np
print(np.eye(2, dtype=int))
    ：[[1 0]
          [0 1]]
print(np.eye(3, k=1))   
    ：[[0. 1. 0.]
          [0. 0. 1.]
          [0. 0. 0.]]