numpy.identity

numpy.identity 함수는 단위행렬 (identity matrix)를 반환합니다.

단위행렬은 주대각선의 원소가 1이고, 나머지는 모두 0인 정사각행렬입니다.


../_images/numpy_identity_01.png

그림. 단위행렬 (Identity matrix).



예제

import numpy as np

print(np.identity(3))
print(np.identity(10)[2:3][0])
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
[0. 0. 1. 0. 0. 0. 0. 0. 0. 0.]

지정한 크기의 단위행렬을 만들거나

슬라이스를 이용해서 특정 위치의 값만 1로 초기화할 수 있습니다.

../_images/numpy_identity_02.png

그림. 단위행렬 (Identity matrix) 슬라이스하기.



이전글/다음글

이전글 :
다음글 :