要創建NumPy的零數組和單位數組,可以使用numpy.zeros()
和numpy.eye()
函數。
import numpy as np
zero_array = np.zeros((3,3))
print(zero_array)
輸出:
[[0. 0. 0.]
[0. 0. 0.]
[0. 0. 0.]]
import numpy as np
eye_array = np.eye(3)
print(eye_array)
輸出:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
這樣就可以創建出指定形狀的零數組和單位數組。