NEW ARTICLE
WinSCP 現在のディレクトリをブックマークするショートカットキー
  1. HOME
  2. python
  3. python numpyで単位行列を作成する

python numpyで単位行列を作成する

python numpyで単位行列を作成する

pythonで、ライブラリnumpyのidentityを使用して、単位行列を作成するサンプルコードを記述してます。pythonのバージョンは3.8.5を使用してます。

目次

環境

  • OS windows10 pro 64bit
  • python 3.8.5

numpyインストール

numpyをインストールされていない方は、pipでインストールしておきます。

pip install numpy

# Successfully installed numpy-1.19.4

identity使い方

identityを使用すると、値が全てゼロになる配列を作成するが可能です。

np.identity(次元数を指定)

以下は、identityを使用して、各次元の単位行列を作成するサンプルコードとなります。

import numpy as np

arr = np.identity(2, dtype=int)

print(arr)

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

arr = np.identity(5, dtype=int)

print(arr)

# [[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]]