import numpy as np
a = np.ones(10)   # 1로 이루어진 크기가 10인 배열 생성
print(a)

[1. 1. 1. 1. 1. 1. 1. 1. 1. 1.]

import numpy as np
a = np.eye(3)  # 3행 x 3열의 단위 배열 생성
print(a)

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]