Diag函数创建矩阵

Question

我将如何创建矩阵

[1  2  0  0  0;
-1  1  2  0  0;
 0 -1  1  2  0;
 0  0 -1  1  2;
 0  0  0 -1  1]

在MatLab中使用diag命令？

Answer 1

这是一种方式：

> diag(ones(1,5),0)+diag(ones(1,4),1)*2+diag(ones(1,4),-1)*-1
ans =

   1   2   0   0   0
  -1   1   2   0   0
   0  -1   1   2   0
   0   0  -1   1   2
   0   0   0  -1   1

> 

这只会在0，+ 1和-1处创建三个对角线，根据需要缩放它们，然后添加它们。