matlab - 如何从 A 向量中获取对角矩阵

Question

我有一个专栏:

0.0677
0.0584
0.0487
0.0453
0.0394

什么指令会得到下面的输出

0.0677   0          0         0          0
0        0.0584     0         0          0
0        0          0.0487    0          0
0        0          0         0.0453     0
0        0          0         0          0.0394

Answer 1

diag 是正常的 MATLAB 解决方案(正如 posdef 所指出的那样)因此

D = diag(vec);

根据需要为您提供一个包含对角线元素的矩阵。

也许在某些应用中更好的方法是创建一个稀疏矩阵，因为对角矩阵非常稀疏。因此，如果您正在执行矩阵乘法，这将极大地帮助减少不必要的操作数量。

n = length(vec);
D = spdiags(vec(:),0,n,n);

如果您真的想以显式形式进行赋值，请使用这样的单个线性索引:

n = length(vec);
D = zeros(n);
D(cumsum([1,repmat(n+1,1,n-1)])) = vec;

或者您可以使用 sub2ind 函数将一组索引转换为单个索引。