用于展平数组的函数 - Javascript

在 JavaScript 中，展平数组是常见的操作，可以将多维数组转换为一维数组。为了实现这个功能，我们可以使用现有的方法，也可以编写自己的函数来解决问题。

使用现有方法

有两种内置的 JavaScript 方法可以用于展平数组：flat()和flatMap()。

• flat(depth)方法可以将多维数组转换为一维数组。可选的参数depth指定要展平的深度级别，默认为 1。如果您希望完全展平数组，则可以提供无限大的深度级别，例如arr.flat(Infinity)。

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

• flatMap(callback)方法在使用传递给它的回调函数展平数组时，还通过回调函数将每个元素映射到一个新数组中。

const arr = [1, 2, 3];
const flatMapArray = arr.flatMap(num => [num * 2]); // [2, 4, 6]

编写自定义函数

如果您想要更多的控制，或者想要手动实现这个功能，可以编写您自己的函数来展平您的数组。

以下是一个示例函数，可以递归地展平任何嵌套数组：

/**
 * 递归展平任何嵌套数组
 * @param {array} arr - 需要展平的数组
 * @returns {array} - 一维数组
 */
function flattenArray(arr) {
  return arr.reduce(function (flat, toFlatten) {
    return flat.concat(
      Array.isArray(toFlatten) ? flattenArray(toFlatten) : toFlatten
    );
  }, []);
}

const nestedArray = [1, 2, [3, 4, [5, 6]]];
const flattenedArray = flattenArray(nestedArray); // [1, 2, 3, 4, 5, 6]

此函数采用 reduce() 方法，它以初始值[]开始遍历数组，并在每个迭代中与当前元素进行连接。当当前元素是一个数组时，它会递归地重复此过程。

结论

展平数组是 JavaScript 中的常见操作之一，很容易使用内置的flat()和flatMap()方法，或编写自己的函数来实现。这可以让我们更轻松地操作嵌套的数组结构。