学习链接

MDN

JavaScript专题之数组扁平化

269个JavaScript工具函数，助你提升工作效率（新）

一文搞定数组扁平化（超全面的数组拉平方案及实现）

数组扁平化

flat 和 flatMap

数组扁平化 - 图1

  1. [1, 2, , 4, 5].flat()
  2. // [1, 2, 4, 5]
  4. [1, 2, [3, [4, 5]]].flat()
  5. // [1, 2, 3, [4, 5]]
  7. [1, 2, [3, [4, 5]]].flat(2)
  8. // [1, 2, 3, 4, 5]
  10. [1, [2, [3]]].flat(Infinity)
  11. // [1, 2, 3]
  1. // 相当于 [[2, 4], [3, 6], [4, 8]].flat()
  2. [2, 3, 4].flatMap((x) => [x, x * 2])
  3. // [2, 4, 3, 6, 4, 8]
  5. // 相当于 [[[2]], [[4]], [[6]], [[8]]].flat()
  6. [1, 2, 3, 4].flatMap(x => [[x * 2]])
  7. // [[2], [4], [6], [8]]

不用 flat 和 flatMap

  1. function flatten(arr) {
  2.     while(arr.some(item => Array.isArray(item))) {
  3.         arr = [].concat(...arr);
  4.     }
  6.     return arr;
  7. }
  1. Array.prototype.myFlat = function (depth = 1) {
  2.     if (depth < 1) return [...this];
  4.     return this.map(val => {
  5.         return [].concat(Array.isArray(val) ? val.myFlat(depth - 1) : val);
  6.     });
  7. }
  1. Array.prototype.myFlat = function (depth = 1) {
  2.     if (depth < 1) return this.slice();
  4.     return this.reduce((arr, val) => {
  5.         return arr.concat(Array.isArray(val) ? val.myFlat(depth - 1) : val);
  6.     }, []);
  7. }
  1. function* flatten(array) {
  2.     for (const item of array) {
  3.         if (Array.isArray(item)) {
  4.             yield* flatten(item);
  5.             continue;
  6.         }
  7.         yield item;
  8.     }
  9. }
  11. const arr = [1, 2, [3, 4, [5, 6]]];
  12. const flattened = [...flatten(arr)];