python 码题集单位矩阵

以下是 Python 代码，用于生成单位矩阵：

def identity_matrix(n):
    """
    生成 n 阶单位矩阵
    """
    matrix = [[0 for j in range(n)] for i in range(n)]
    for i in range(n):
        matrix[i][i] = 1
    return matrix

其中，`n` 表示矩阵的阶数，`matrix` 用于存储生成的矩阵。首先，我们定义一个 `n` 行 `n` 列的矩阵，初始化所有元素为 0。然后，对角线上的元素设置为 1，即为单位矩阵。最后，返回生成的单位矩阵。