numpy 新建单位阵

可以使用 numpy.identity() 函数来创建一个新的单位矩阵，该矩阵的大小由参数指定。例如，创建一个 3x3 的单位矩阵可以使用以下代码：

import numpy as np

identity_matrix = np.identity(3)
print(identity_matrix)

输出结果为：

array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

其中，np.identity(3) 表示创建一个大小为 3x3 的单位矩阵。