How do I insert multiple vectors as diagonals in the same matrix?

74 次查看(过去 30 天)
I'm trying to use a forward-difference method to solve a parabolic PDE. I need a matrix with the main diagonal and both diagonals one spot off from the main diagonal to be defined and the rest 0. My current code is this:
v = lambda*ones(1,m-2);
v1 = (1-2*lambda)*ones(1,m-1);
A = diag(v,1);
disp(A);
A = diag(v1,0);
disp(A);
A = diag(v,-1);
disp(A);
lambda is just a scalar. This is what my intuition told me to do, but after each time I do A=diag, it overwrites the previous matrix. I need the vector v to be on the off diagonals -1 and 1, and the vector v1 to be on the main diagonal. This is probably pretty easy but I am just missing it. Any help would be appreciated.

采纳的回答

Kelly Kearney
Kelly Kearney 2019-4-16
You're close... you just need to add the matrices together:
A = diag(v,1) + diag(v,-1) + diag(v1,0)

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Geometry and Mesh 的更多信息

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by