Chunk

原创 codeniu 2023-06-15

283

源码

创建数组按指定长度分组

// Underscore/Lodash
_.chunk(['a', 'b', 'c', 'd'], 2);
// => [['a', 'b'], ['c', 'd']]

_.chunk(['a', 'b', 'c', 'd'], 3);
// => [['a', 'b', 'c'], ['d']]

// Native

const chunk = (input, size) => {
  return input.reduce((arr, item, idx) => {
    return idx % size === 0
      ? [...arr, [item]]
      : [...arr.slice(0, -1), [...arr.slice(-1)[0], item]];
  }, []);
};

chunk(['a', 'b', 'c', 'd'], 2);
// => [['a', 'b'], ['c', 'd']]

chunk(['a', 'b', 'c', 'd'], 3);
// => [['a', 'b', 'c'], ['d']]

源码解析

chunk 函数接受一个数组 input 和一个大小 size 作为参数。函数返回一个新的数组，其中每个子数组的最大长度为 size（除了可能是最后一个子数组，它可能会更短）。

该函数使用 reduce 方法遍历 input 数组。对于索引为 idx 的每个元素 item，它检查 idx 是否是 size 的倍数。如果是，将创建一个新的子数组，其中只包含 item，并将其添加到结果数组 arr 中。如果不是，则删除 arr 中的最后一个子数组，将 item 添加到其中，并将修改后的子数组再次添加到 arr 中。

通过这种方式，input 数组被拆分为长度为 size 或更短的子数组。