用 ES6 Array.prototype.flat 函数实现数组降维

2024-01-21 23:17:37

前言

数组降维，又称数组扁平化，是指将多维数组转换为一维数组的过程。数组降维在实际开发中经常会遇到，比如需要将一个包含子数组的数组转换为一个包含所有元素的数组，或者需要将一个多维数组转换为一个适合存储在数据库中的一维数组。

ES6 Array.prototype.flat 函数

ES6 中的 Array.prototype.flat 函数可以实现数组降维。该函数接受一个参数，指定要降维的层数。如果省略该参数，则默认将数组降维一层。

const arr = [1, [2, [3, 4]]];
console.log(arr.flat()); // [1, 2, 3, 4]
console.log(arr.flat(2)); // [1, 2, 3, 4]

从上面的示例可以看出，flat 函数可以将多维数组降维为一维数组。如果指定了降维的层数，则该函数会将数组降维到指定层数。

递归方法

除了使用 flat 函数之外，还可以使用递归的方法实现数组降维。递归的方法是通过不断地将子数组中的元素添加到主数组中，直到所有的子数组都处理完。

function flatten(arr) {
  let result = [];
  for (let i = 0; i < arr.length; i++) {
    if (Array.isArray(arr[i])) {
      result = result.concat(flatten(arr[i]));
    } else {
      result.push(arr[i]);
    }
  }
  return result;
}

const arr = [1, [2, [3, 4]]];
console.log(flatten(arr)); // [1, 2, 3, 4]