JavaScript 实现数组扁平化的常见方法

309 阅读1分钟

在处理数据时,经常会遇到需要将多维数组转化成一维数组的场景,也就是数组扁平化,这里我总结了三种方式来实现数组扁平化。

1. reduce + 递归

使用reduce处理数组,reduce的第一个参数是一个函数,第二个参数是可选的,是传递给函数的初始值。如果没有提供初始值,则将使用数组中的第一个元素

这里传入一个空数组[]作为初始值,接着判断cur是否是数组,若为数组,则递归调用;若不是数组,就将cur拼接到pre

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

const flatten = arr => {
    return arr.reduce((pre, cur) => {
        return pre.concat(Array.isArray(cur) ? flatten(cur) : cur);
    }, []);
};

const res = flatten(arr);
console.log(res); // [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

2. 直接用flat()

flat()方法会按照一个可指定的深度递归遍历数组,并将所有元素与遍历到的子数组中的元素合并为一个新数组返回。

参数为指定要提取嵌套数组的结构深度,可选,默认为1

const flatten = arr => arr.flat(Infinity);

3. while循环

利用展开运算符,可以将二维数组转化为一维数组:

[].concat(...[1, 2, [3, 4]]); // [1, 2, 3, 4]

再利用while循环,若arr出现数组,则使用一次展开运算符,直到没有为止。

const flatten = arr => {
    while (arr.some(item => Array.isArray(item))) {
        arr = [].concat(...arr);
    }
    return arr;
};