可以通过使用reduce
方法和Map
数据结构来解决此问题。
首先,我们需要将需要分组的属性作为一个数组传递给方法。例如,如果我们想按year
和month
分组,则需要传递['year', 'month']
。
然后,我们使用reduce
方法遍历数组,并将每个组合的属性值存储在一个字符串中作为Map数据结构的键。如果该键不存在,则需要初始化一个新的组,并将其添加到Map中。
在处理完所有项后,我们需要从Map的值中获取一个数组,并对数组中的所有数字进行求和。
以下是代码示例:
const data = [
{ year: 2018, month: 1, value: 100 },
{ year: 2018, month: 1, value: 200 },
{ year: 2018, month: 2, value: 300 },
{ year: 2019, month: 1, value: 400 }
];
function sumByMultiGroup(data, keys) {
return data.reduce((acc, item) => {
const key = keys.map(k => item[k]).join('-');
if (!acc.has(key)) {
acc.set(key, []);
}
acc.get(key).push(item.value);
return acc;
}, new Map());
}
const groupResult = sumByMultiGroup(data, ['year', 'month']);
// 打印结果:Map(3) { '2018-1' => [ 100, 200 ], '2018-2' => [ 300 ], '2019-1' => [ 400 ] }
console.log(groupResult);
// 对分组结果求和
const sumResult = {};
for (let [key, values] of groupResult) {
sumResult[key] = values.reduce((a, b) => a + b, 0);
}
// 打印结果: { '2018-1': 300, '2018-2': 300, '2019-1': 400 }
console.log(sumResult);