numpy.identity()

動作

このメソッドは，単位行列を生成します．生成される行列は正方行列です．

引数と戻り値

引数は「identity(n[dtype])」です．n: 配列の行数・列数です．dtype はデータの型 (default folat64) です．

戻り値は，Numpy の ndarray です．

プログラム例

import numpy as np

a = np.identity(2)
b = np.identity(3, dtype=np.int)

print('a = ', a, '\n')
print('b = ', b, '\n')

実行結果

a =  [[ 1.  0.]
      [ 0.  1.]] 

b =  [[1 0 0]
      [0 1 0]
      [0 0 1]]