ES8 中新增了 Array.prototype.flatMap() 方法，它基本上是 Array.prototype.map() 和 Array.prototype.flat() 方法的组合。这个方法非常有用，可以用来将嵌套的数组展平，然后再对每个元素进行操作。然而，在使用这个方法的时候，你可能会遇到一些值类型问题。本文将介绍这些问题，并提供解决方案。
问题描述
在使用 Array.prototype.flatMap() 方法时，你可能会遇到以下两种情况：
空元素被省略
如果一个嵌套数组中包含一个空元素，它将被省略。例如，下面的代码：
const arr = [1, 2, [3, 4, , 5], 6]; const flatArr = arr.flatMap((x) => x + 1); console.log(flatArr);
输出结果为：
[2, 3, 4, 5, 7]
可以看到，空元素已经被省略了。
数字被转换成字符串
在使用 Array.prototype.flatMap() 方法时，数字会被自动转换成字符串。例如，下面的代码：
const arr = [1, 2, [3, 4, 5], 6]; const flatArr = arr.flatMap((x) => [x, x ** 2]); console.log(flatArr);
输出结果为：
[1, 1, 2, 4, 3, 9, 4, 16, 5, 25, 6, 36]
可以看到，数字已经被转换成了字符串。这可能会导致一些问题。
解决方案
空元素被省略
为了解决省略空元素的问题，我们可以定义一个自定义 flatMap() 方法，这个方法可以将空元素保留。以下是这个方法的代码：
Array.prototype.customFlatMap = function (callback) { return this.reduce((acc, curr) => { const res = callback(curr); return res ? [...acc, ...res] : acc.concat(curr); }, []); };
现在，我们可以使用这个自定义 flatMap() 方法来解决空元素被省略的问题。例如，下面的代码：
const arr = [1, 2, [3, 4, , 5], 6]; const flatArr = arr.customFlatMap((x) => x + 1); console.log(flatArr);
输出结果为：
[2, 3, 4, , 5, 7]
可以看到，空元素已经被保留了。
数字被转换成字符串
为了解决数字被转换成字符串的问题，我们可以使用显式类型转换将数字转换成字符串。以下是这个方法的代码：
-- -------------------- ---- ------- ----------------------------- - -------- ---------- - ------ ----------------- ----- -- - ----- --- - --------------- ------ --- - -------- ------- - ----------------- -- ---- -- ----- --- - --- -- --- -- --- --- ----- ------- - --------------------- -- ----------- -------- -- ----- ---------------------展开代码
输出结果为：
["1", "1", "2", "4", "3", "9", "4", "16", "5", "25", "6", "36"]
可以看到，数字没有被转换成字符串了。
结论
在使用 ES8 中的 Array.prototype.flatMap() 方法时，你可能会遇到一些值类型问题。但是，我们可以通过自定义 flatMap() 方法以及显式类型转换的方法来解决这些问题。这些方法将帮助你更好地使用这个方法，并获得更好的结果。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/673848e6317fbffedf0f521e
