numpy

  numpy的eye用于生成单位矩阵：

import numpy as np

# 创建3阶单位矩阵
a_eye = np.eye(3)
print(a_eye)

执行结果：

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