numpy.identity¶

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

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

그림. 단위행렬 (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로 초기화할 수 있습니다.

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