julia - 在 Julia 中对整数数组求和的有效方法

Question

我有一个 2D 数组，我想修改它以便将给定的元素与它之前的所有元素相加，例如，如果我有一个数组:

[1 2; 3 6; 4 7; 4 8]

我希望能够将其转换为

[1 2; 4 8; 8 15; 12 23]

我可以在 julia 中使用以下代码段来做到这一点:

for i in 1:10,
   for k in 2:size(d,1),
          d([k,i] += d[k-1,i)];
   end
end

但我认为必须有更有效的方法来做到这一点？

Answer 1

是的，有:累积

julia> d = [1 2; 3 6; 4 7; 4 8]
4x2 Array{Int64,2}:
 1  2
 3  6
 4  7
 4  8

julia> cumsum(d)
4x2 Array{Int64,2}:
  1   2
  4   8
  8  15
 12  23