对数据进行分组,类似sql的groupBy。
原理:遍历数组,将符合条件的数据放在一起,最后返回一个分组后的二维数组。
//分组
export const groupBy = (list, fn) => {
const groups = {};
list.forEach(function (o) {
const group = JSON.stringify(fn(o));
groups[group] = groups[group] || [];
groups[group].push(o);
});
// return Object.keys(groups).map(function (group) {
// return groups[group];
// });
return groups;
}
list:原始数据。
fn:分组条件判断函数,之后会根据该函数返回的结果进行分组,其有一个参数表示数组的某一项数据。
示例:
let links = [
{ source: 'test1', target: 'test1', value: 10 },
{ source: 'test1', target: 'test2', value: 30 },
{ source: 'test1', target: 'test3', value: 40 },
{ source: 'test1', target: 'test4', value: 20 }
]
let groupData = groupBy(links, (link) => {
return link.source
})
console.log(groupData)
// 返回结果
//{
// "test1":[
// { source: "test1", target: "test1", value: 10 },
// { source: "test1", target: "test2", value: 30 },
// { source: "test1", target: "test3", value: 40 },
// { source: "test1", target: "test4", value: 20 }
// ]
// }