MATLAB: How to create the below Matrix

nxn matric diag function

please see the attached image i need to create a tridiagonal nxn matrix using the diag function. i know how to make it this way but thats not what is required.
>> A=[2 -1 0 0 0;-1 2 -1 0 0;0 -1 2 -1 0;0 0 -1 2 -1; 0 0 0 -1 2]
A =
     2    -1     0     0     0
    -1     2    -1     0     0
     0    -1     2    -1     0
     0     0    -1     2    -1
     0     0     0    -1     2

Best Answer

Hi,
As James wrote you:Have you reviewed the doc on diag to see how to get values into the off-diagonals?
Mathworks offers an excellent documentation for such topics with plenty of examples.
You could use this
    function Out = mydiag(n)
     %n size of matrix Out
     Out= diag(-1*ones(1,n-1),-1) + diag(2*ones(1,n)) + diag(-1*ones(1,n-1),1);
    end
>> mydiag(5)
ans =
       2    -1     0     0     0
      -1     2    -1     0     0
       0    -1     2    -1     0
       0     0    -1     2    -1
       0     0     0    -1     2
Related Question