在Python3中使用矩阵作为对角元素并使用恒等矩阵作为上下对角矩阵来构建三对角矩阵

Question

我有两组矩阵，一组是对角矩阵（Size = n），另一组是单位矩阵（Size = n）。我必须制作一个三对角矩阵，使我的对角矩阵作为（n ^ 2 ）大小矩阵和单位矩阵分别作为上下对角线。 Matrix should be in this form