Array - groupBy

2019-04-30

描述

基于给定的函数，对数组中的元素进行分组。

提示

• 使用 Array.prototype.map() 迭代数组中的每一个元素，返回函数运行的结果或其属性值
  to map the values of an array to a function or property name.
• 使用 Array.prototype.reduce() 创建一个对象，他的 key 来自于 map 计算后的结果

代码

const groupBy = (arr, fn) =>
  arr.map(typeof fn === 'function' ? fn : val => val[fn]).reduce((acc, val, i) => {
    acc[val] = (acc[val] || []).concat(arr[i]);
    return acc;
  }, {});

示例

按下取整结果进行分类：

groupBy([6.1, 4.2, 6.3], Math.floor); // {4: [4.2], 6: [6.1, 6.3]}

按字符长度进行分类：

groupBy(['one', 'two', 'three'], 'length'); // {3: ['one', 'two'], 5: ['three']}

返回总目录

每天 30 秒系列之 JavaScript 代码