计算矩阵的每个对角线的总和

Question

我有一个像A这样的矩阵,我想计算这个矩阵的每个对角线的总和,并在像Y这样的向量中显示它.

A=[1 2 3; 4 5 6; 7 8 9]

Y=[3 8 15 12 7]

我知道代码

[sum(diag(y,2)),sum(diag(y,1)),sum(diag(y,0)),sum(diag (y,-1)),sum(diag (y,-2))]

但我想把它写成一个函数.

Answer 1

spdiags 可以做你想要的:

dsum = sum(spdiags(A))

您可以使用反转向量fliplr并创建一个函数:

function dsum = diagsum(A)
    dsum = fliplr(sum(spdiags(A)));
end

结果

dsum =

    3    8   15   12    7