Python生成一个单位矩阵

可以使用 NumPy 库中的 `eye` 函数来生成一个单位矩阵。代码如下：

import numpy as np

n = 3  # 矩阵的维度
I = np.eye(n)  # 生成一个 n x n 的单位矩阵
print(I)

输出结果为：

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