matlab - 在 Matlab 中执行此矩阵运算是否更有效

Question

我想在 Matlab 中做以下事情:给定一个矩阵 H，我想构建一个相同大小的矩阵 H*，使得 H*(:,i) 是下一列的总和(即 i+1 -> n) 的 H。例如，如果 H 是

H =

 2     4     7    14
 3     5    11    -3

我希望 H* 是

25    21    14     0
13     8    -3     0

到目前为止，我已经完成了以下代码，但它涉及一个 for 循环，所以我不希望它非常有效(特别是，我的矩阵将有大量的列在实际应用中我会用到)。

H_tilde=zeros(size(H));
for i=1:size(H,2)
    H_tilde(:,i)=sum(H(:,i+1:size(H,2)),2);
end

有没有办法让它变得更好？

Answer 1

使用cumsum沿第 2 个维度使用 'reverse' 选项，如下所示:

H_tilde = [cumsum(H(:, 2:end), 2, 'reverse') zeros(size(H,1), 1)];