JavaScript 扁平化多維陣列

發表於 | 更新於 | 分類於

前言

今天遇到一個需求需要將多維陣列扁平化，這邊紀錄一下我查到的資料該怎麼做，順便記錄一下今天看到的 ES 2019 的方式

多維陣列

一開始我們會持有一組多維陣列，何謂多維陣列呢?簡單來講就是陣列中還有陣列

讓我們來看這一次的範例

1
const a = ['1',['2','3'],['4','5'],['6','7']];

上面這個範例就是一個多維陣列，那扁平化是什麼意思?簡單來講就是我們會希望多維陣列結果是這樣

1
const b = ['1', '2', '3', '4', '5', '6', '7'];

第一種方法

1
2
3
const a = ['1',['2','3'],['4','5',['6','7']]];
const a = ['1',['2','3'],['4','5'],['6','7']];
const newArray = [].concat.apply([], a);

concat + apply

第二種方法

1
2
3
4
5
const a = ['1',['2','3'],['4','5'],['6','7']];

const newArray = a.reduce((a, b) => {
  return a.concat(b);
})

這個方式會得到這樣的結果
reduce

但是如果變數是這樣

1
2
3
4
5
const a = [['2','3'],['4','5'],['6','7']];

const newArray = a.reduce((a, b) => {
  return a.concat(b);
})

那就會得到正常的陣列扁平化

Image

第三種方法

1
2
3
4
5
6
7
8
const a = ['1',['2','3'],['4','5'],['6','7']];

const array = [];
const newArray = a.forEach((item, index) => {
  item.forEach((b) => {
    array.push(b);
  })
})

這時候一樣會得到一個錯誤

forEach

如果變數改成這樣

1
2
3
4
5
6
7
8
const a = [['2','3'],['4','5'],['6','7']];

const array = [];
const newArray = a.forEach((item, index) => {
  item.forEach((b) => {
    array.push(b);
  })
})

那就一樣可以取得扁平化後的陣列

push

第四種方法

1
2
3
const a = ['1',['2','3'],['4','5'],['6','7']];

const newArray = [].concat(...a);

使用 ES 6 展開的方法就可以很輕鬆地做到我們要的需求

展開

第五種方法

第五種方法比較新，這是使用了 ES 2019 的新方法

1
2
3
const a = ['1',['2','3'],['4','5'],['6','7']];

const newArray = a.flat(Infinity);

flat

flat() 可以參考這篇

文章參考

在 JavaScript 中將多維陣列扁平化
Array.prototype.flat and Array.prototype.flatMap

補充

這邊補上友站宇軒給的遞迴範例與解答~

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
// 練習二：壓平陣列
// 請寫出一個 function flatten，接收一個多維陣列並回傳「壓平」後的陣列。
// flatten([1, 2, 3]) 預期回傳值：[1, 2, 3]
// flatten([1, 2, [1, 2], [1, 3], 6]) 預期回傳值：[1, 2, 1, 2, 1, 3, 6]
// flatten([1, [2, [3, [4]], 5], 6]) 預期回傳值：[1, 2, 3, 4, 5, 6]
// 遞迴
function flatten(array) {
  let tempArray = [];
  for (let i = 0; i < array.length; i++) {
    if (Array.isArray(array[i])) {
      // 壓平陣列 CALL 自己
      let flat = flatten(array[i]); // ex [1, 2, 3]
      // 推入陣列
      for (let j = 0; j < flat.length; j++) {
        tempArray.push(flat[j]);
      }
    } else {
      tempArray.push(array[i]);
    }
  }
  return tempArray;
}
0%