matlab - 我需要创建一个对角矩阵，其中对角元素是一个向量

Question

矢量应该看起来像，

[1 2 3 0 0 0
 0 1 2 3 0 0
 0 0 1 2 3 0
 0 0 0 1 2 3];

我知道我希望沿对角线“粘贴”的向量 ([1 2 3])，但我不知道数组的大小，因此行数需要由变量 N 确定。

Answer 1

您可以使用 spdiags设置对角线并具有所需的形状:

n = 4;
A = full(spdiags(ones(n,1)*[1,2,3],[0,1,2],n,n+2));

返回:

A =

   1   2   3   0   0   0
   0   1   2   3   0   0
   0   0   1   2   3   0
   0   0   0   1   2   3