파이썬 넘파이 대각행렬(diagonal matrix), 항등행렬(identity matrix). np.diag, np.identity, np.eye

파이썬 넘파이 대각행렬(diagonal matrix), 항등행렬(identity matrix).¶

In [1]:

import numpy as np

In [2]:

# diagonal matrix
np.diag([4,2,6,3,6])

Out[2]:

array([[4, 0, 0, 0, 0],
       [0, 2, 0, 0, 0],
       [0, 0, 6, 0, 0],
       [0, 0, 0, 3, 0],
       [0, 0, 0, 0, 6]])

In [3]:

#identity matrix
np.identity(5)

Out[3]:

array([[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.]])

In [4]:

np.eye(3)

Out[4]:

array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])