JavaScript之数组扁平化——巧用flat函数平复你的数组

2023-09-17 09:58:30

踏入编程世界，总免不了与各种数据结构打交道，其中数组尤为常见。当我们需要处理多维数组时，数组扁平化便成了必不可少的技能。所谓数组扁平化，就是将多维数组中的元素“拉平”，使其成为一个一维数组。这样一来，数组结构更加清晰，便于操作和处理。

在JavaScript中，数组扁平化可以借助flat函数轻松实现。flat函数是ES2019中引入的新特性，它可以将数组中的所有元素递归地展平到指定深度。使用flat函数，我们可以轻松将多维数组转换为一维数组。例如：

const multidimensionalArray = [
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
];

const flattenedArray = multidimensionalArray.flat();

console.log(flattenedArray); // 输出：[1, 2, 3, 4, 5, 6, 7, 8, 9]

简单几行代码，我们就将一个三维数组扁平化成了一个一维数组。是不是很方便呢？

flat函数还可以接受一个参数，指定展平的深度。例如，如果我们只想展平一层，可以使用以下代码：

const flattenedArray = multidimensionalArray.flat(1);

console.log(flattenedArray); // 输出：[1, 2, 3, 4, 5, 6, 7, 8, 9]

是不是一如既往的简单？

除了flat函数，我们还可以使用递归的方式实现数组扁平化。递归是一种函数自调用的技术，通过不断调用自身来解决问题。

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

const multidimensionalArray = [
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
];

const flattenedArray = flattenArray(multidimensionalArray);

console.log(flattenedArray); // 输出：[1, 2, 3, 4, 5, 6, 7, 8, 9]