Javascript中的多维数组累积和

Multi dimensional array cumulative sum in Javascript

本文关键字：数组 Javascript      更新时间：2023-09-26

给定一个多维数组：

var a = [[3,2,5], [4,1,7], [1,6,8]];

我想对每个数组进行累积求和，以返回以下结果：

[[3,2,5], [7,3,12], [8,9,20]];
  • 每个子数组的第一个元素的总和：3 4 1
  • 每个子数组的第二个元素的总和：2 1 6
  • 每个子阵列的第三个元素的总和：5 7 8

我试过使用reduce（），但不能完全得到预期的结果。

任何建议，不胜感激。

S

更新-将其提升到下一个层次：

var a = [
    [new Date(), 3,2,5], 
    [new Date(), null,1,7], 
    [new Date(), null,6,8], 
    [new Date(), 1,2,3]
];

应导致：

[[new Date(), 3,2,5], 
 [new Date(), null,3,12],
 [new Date(), null,9,20],
 [new Date(), 4,11,23]]

我的方法是创建一个多维offsetIndex数组：

var offsetIdx = [];
        for (var i=1; i<a.length; i++) {
            for (var z=0; z<a[i].length; z++) {
                var zValue = a[i][z];
                oIdx = offsetIdx[z] || 0;
                a[i][z] = zValue && z!==0 ? a[i-1-oIdx][z] + zValue : zValue;
                if(!zValue){
                    offsetIdx[z] = oIdx + 1;
                } else {
                    offsetIdx[z] = 0; 
                }
            }
        }

很高兴看到其他方法和方法使其超轻。

for (var i=1; i<a.length; i++) {
  for (var z=0; z<a[i].length; z++) {
   a[i][z] = a[i-1]][z] + a[i][z]
  }
}

数组应该随着循环的运行而动态更新。这是破坏性的，因此它将修改原始数组。

  function cumulativeSum(arr) {
    var result = [arr[0]];
    for(var i = 1; i < arr.length; i++) {
        result.push([]);
        for(var j = 0; j < arr[0].length; j++) {
          result[i].push(result[i - 1][j] + arr[i][j]);
        }
    }
    return result;
}
    
document.body.innerHTML = JSON.stringify(cumulativeSum(
  [[3,2,5], [4,1,7], [1,6,8]]
))

与另一个答案不同，这个答案没有破坏性，保留了原始数组并返回结果。

使用Array.reduce，它看起来像这个

var arr  = [[3,2,5], [4,1,7], [1,6,8]];
var arr2 = arr.reduce(function(a,b) {
    var nested = Array.isArray(a[0]);
    b = b.map(function(x,i) {
    	return x + (nested ? a[a.length-1] : a)[i];
    });
    if ( nested ) a.push(b);
    return nested ? a : [a,b];
});
document.body.innerHTML = '<pre>' + JSON.stringify(arr2, 0, 4) + '</pre>';

这是一个稍微"优化"的（golfed）版本，传递了一个减少和切片阵列的起点

var arr  = [[3,2,5], [4,1,7], [1,6,8]];
var arr2 = arr.slice(1).reduce(function(a,b) {
	return [a.push(b.map(function(x,i) {return x+a[a.length-1][i]})), a].pop();
},[arr[0]]);
document.body.innerHTML = '<pre>' + JSON.stringify(arr2, 0, 4) + '</pre>';

使用ES2015 使其成为一条直线

var arr  = [[3,2,5], [4,1,7], [1,6,8]];
var a2 = arr.slice(1).reduce((a,b)=>[a,a.push(b.map((x,i)=>x+a[a.length-1][i]))][0],[arr[0]]);
document.body.innerHTML = '<pre>' + JSON.stringify(a2, 0, 4) + '</pre>';

这将为您提供所有元素的总和。不完全是你所问的，但我会把这个答案留在这里，供未来的访问者了解问题的标题。

  1. 先用你喜欢的库压平（下划线和lodash都有）

  2. 然后减少+求和。

    _.flatten([1, [2, [3, [4]], 5]]);
// → [1, 2, [3, [4]], 5]

真是一道数学题。

为什么不先转置数组？这个问题的答案-在JavaScript中转换2D数组-建议使用undercore.js解决方案：

_.zip.apply(_, [[1,2,3], [1,2,3], [1,2,3]])

这样做。有很多方法。

那么求和应该容易得多——只需要.map(f)，其中f是一个数组函数的和。

IMO这是一个很好且可读的解决方案，因为"转置+求和"非常符合问题的列和性质，我会避免使用命令式或循环式的解决方案来掩盖这一点。

相关文章：