Javascript实现数据分组功能，高效实现按照指定属性分组聚合

小蓝 • 2024-11-19 19:00 • 编程

一、什么是数据分组？

在数据分析和处理中，我们常常需要按照某些属性对数据进行聚合操作。以一个包含多个对象的数组为例，我们可以按照对象的某个属性，将数组分成若干个小数组，每个小数组包含具有相同属性值的对象，这就是数据分组的操作。

例如，下面是一个包含若干个对象的数组：

const users = [
  {id:1, name:'张三', age:20, gender:'男'},
  {id:2, name:'李四', age:22, gender:'男'},
  {id:3, name:'王五', age:23, gender:'女'},
  {id:4, name:'赵六', age:25, gender:'男'},
  {id:5, name:'钱七', age:22, gender:'女'}
];

我们可以按照年龄属性进行分组：

// 按照年龄分组
const groups = groupBy(users, 'age');
console.log(groups);
/* 输出
{
  "20": [
    { "id": 1, "name": "张三", "age": 20, "gender": "男" }
  ],
  "22": [
    { "id": 2, "name": "李四", "age": 22, "gender": "男" },
    { "id": 5, "name": "钱七", "age": 22, "gender": "女" }
  ],
  "23": [
    { "id": 3, "name": "王五", "age": 23, "gender": "女" }
  ],
  "25": [
    { "id": 4, "name": "赵六", "age": 25, "gender": "男" }
  ]
}
*/

按照年龄属性分组后，返回的对象中，每个小数组包含了具有相同年龄的对象。

二、如何高效实现数据分组？

实现数据分组的方法有很多种，但是如何高效实现数据分组呢？下面介绍两种高效的实现方法，分别是使用reduce方法和Map对象。

三、使用reduce方法实现数据分组

reduce方法可以对数组中的每个元素进行聚合操作。我们可以使用reduce方法，对每个属性值进行分组聚合。

/**
 * 按照指定属性对数据进行分组聚合
 * @param {Array} data - 数据数组
 * @param {String} key - 属性名
 * @returns {Object} - 分组聚合后的对象
 */
function groupBy(data, key) {
  return data.reduce((result, item) => {
    const groupKey = item[key];
    if (!result[groupKey]) {
      result[groupKey] = [];
    }
    result[groupKey].push(item);
    return result;
  }, {});
}

使用reduce方法实现数据分组的核心代码就是上面的函数。这个函数接受两个参数，一个是数据数组，一个是属性名。reduce方法对每个元素进行聚合操作，这里使用了一个result变量来保存聚合结果，在遍历每个元素时，首先获取元素的属性值，然后判断这个属性值是否已经存在于result变量中，如果不存在，则新建一个数组，并将这个数组加入到result变量中，如果存在，则直接将元素加入到对应的数组中，最后返回result变量。

四、使用Map对象实现数据分组

Map对象是ES6中的一个新的数据结构，它接受一个可迭代对象作为参数，可以将其中每个元素作为键值对存储在Map对象中，并提供了一些方法用于操作这些键值对。

/**
 * 按照指定属性对数据进行分组聚合
 * @param {Array} data - 数据数组
 * @param {String} key - 属性名
 * @returns {Object} - 分组聚合后的对象
 */
function groupBy(data, key) {
  const groups = new Map();
  data.forEach(item => {
    const groupKey = item[key];
    if (!groups.has(groupKey)) {
      groups.set(groupKey, []);
    }
    groups.get(groupKey).push(item);
  });
  return Object.fromEntries(groups);
}

使用Map对象实现数据分组的核心代码就是上面的函数。这个函数接受两个参数，一个是数据数组，一个是属性名。首先创建一个空的Map对象，然后遍历数据数组，对每个元素进行操作，获取属性值，判断这个属性值是否已经存在于Map对象中，如果不存在，则新建一个空数组，并将这个数组存入Map对象中，如果存在，则直接将元素加入到对应的数组中，最后将Map对象转化为一个普通对象，返回即可。

五、结语

使用reduce方法和Map对象，我们可以高效地实现数据分组操作。在实际开发中，数据分组是一个非常常见的操作，如果我们掌握了高效实现数据分组的方法，可以更便捷地处理和分析数据，提高代码的效率和可维护性。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/159155.html