python生成单位阵

生成n阶单位矩阵的Python代码如下：

n = 5  # 矩阵阶数
identity_matrix = [[0] * n for i in range(n)]  # 创建n x n的零矩阵
for i in range(n):
    identity_matrix[i][i] = 1  # 将对角线元素设为1
print(identity_matrix)

输出结果为：

[[1, 0, 0, 0, 0], 
 [0, 1, 0, 0, 0], 
 [0, 0, 1, 0, 0], 
 [0, 0, 0, 1, 0], 
 [0, 0, 0, 0, 1]]

其中，`identity_matrix`就是所求的单位矩阵。