记一次JavaScript数组扁平化学习

在写项目的时候遇到了多级目录的树状结构，需要将这些嵌套的数组转换成只有一层的数组进行展示，因而有了这次数组扁平化常用方法的学习总结，在此记录。

作者：Mia不是大米虫
链接：https://juejin.im/post/5ed52d0ee51d45788325f7a0

什么是数组扁平化？

数组扁平化是指将多维数组转换为一维数组，即将内部嵌套着别的数组的数组提炼成没有嵌套的一维数组。举个栗子：

// 原数组：

var array = [1, [[2, 3], 4]]

// 目标数组：

var array = [1, 2, 3, 4]

复制代码

如何实现数组扁平化？

Array.prototype.flat()

这是ES6提供的方法，用于数组扁平化，这个方法返回一个新的数组，并不会改变原数组：

let arr1 = [1, [2, 3], 4]

console.log(arr1.flat())

// 打印[1, 2, 3, 4]

复制代码

值得注意的是 flat() 默认扁平一层嵌套：

let array = [1, [[2, 3], 4]]

console.log(array.flat())

// 打印[1, [2, 3], 4]

复制代码

可以带一个整数参数表示扁平的层数：

let array = [1, [[2, 3], 4]]

console.log(array.flat(2))

// 打印[1, 2, 3, 4]

复制代码

如果对于无论多少层嵌套的数组都想转成一维数组，可以用Infinity关键字作为参数：

let array = [1, [[2, 3], 4]]

console.log(array.flat(Infinity))

// 打印[1, 2, 3, 4]

复制代码

使用扩展运算符和concat()

可以使用扩展运算符可以展开数组，再用concat()可以合并数组，但只操作一次的结果只能展开一层：

let array = [1, [[2, 3], 4]]

function flatten(array) {

return [].concat(...array)

}

console.log(flatten(array))

// 打印[1, [2, 3], 4]

复制代码

要完全展开需要对嵌套的数组遍历再进行展开：

let array = [1, [[2, 3], 4]]

function flatten(array){

// 先用some()检测数组中的元素是否为一个数组

while(array.some(item => {

return Object.prototype.toString.call(item) === '[object Array]'

})){

array = [].concat(...array);

}

return array;

}

console.log(flatten(array));

// 打印[1, 2, 3, 4]

复制代码

递归

定义目标空数组，然后我们对原有数组的每一项进行处理，若这一项为数组，则用concat()将这个子项数组合并到目标数组，若这一项不为数组，则直接push到目标数组里。

let array = [1, [[2, 3], 4]]

function flatten(array){

let res = [];

array.forEach((item, index, array) => {

if (Object.prototype.toString.call(item) === '[object Array]'){

res = res.concat(flatten(item));

} else {

res.push(array[index])

}

})

return res;

};

console.log(flatten(array))

// 打印[1, 2, 3, 4]

复制代码

toString()

当元素是字符串或者是数字的时候，toString()可以把数组转为字符串，然后用split()就可以重新获得一个数组，但这个方法是有受限条件的，所以要酌情使用。下面举例一个全是数字的数组使用toString()扁平化：

let array = [1, [[2, 3], 4]]

function flatten(array) {

return array.toString().split(',').map(item => +item);

}

console.log(flatten(array))

// 打印[1, 2, 3, 4]

复制代码

写在最后

这就是本次记录学习数组扁平化的几个方法，数组扁平化的方法还有很多，可能还可以用for循环等方法实现，但Array.prototype.flat()是最方便的hhh。

本文中若是有错误的或者不正当的文字叙述，请告知，谢谢~

参考链接

ECMAScript 6 入门

JS数组专题1️⃣ ➖ 数组扁平化

推荐阅读：

公众号后台回复【前端】加入群

好文和朋友一起看~