es6中数组的flat()和flatMap()


今天看到一道题,把一个嵌套的数组扁平化于是想到es6有这么一段操作

Array.prototype.flat()用于将嵌套的数组“拉平”,变成一维数组。该方法返回一个新数组,对原数据没有影响

而且flat()默认只会“拉平”一层,如果要处理多层的嵌套数组,可以将flat()方法的参数写成一个整数

[1, 2, [3, [4, 5]]].flat()
// [1, 2, 3, [4, 5]]
[1, 2, [3, [4, 5]]].flat(2)
// [1, 2, 3, 4, 5]

如果是不管多少层嵌套,都要转成一维数组,可以用Infinity关键字作为参数,如果有空位,flat()会跳过

[1, [2, [3]]].flat(Infinity)
// [1, 2, 3]
[1, 2, , 4, 5].flat()
// [1, 2, 4, 5]

flatMap()方法对原数组的每个成员执行一个函数,相当于执行Array.prototype.map(),然后对返回值组成的数组执行flat()方法。返回一个新数组,不改变原数组。flatMap()只能展开一层数组

// 相当于 [[2, 4], [3, 6], [4, 8]].flat()
[2, 3, 4].flatMap((x) => [x, x * 2])
// [2, 4, 3, 6, 4, 8]