ES2019でArray#flatMapが導入されたので、それを使ってどんな深さの配列でも1次元にする方法を紹介します。
flatMapの仕様
flatMapはmap()してからflat()するメソッドです。
[1,2,3].map(n => [n,-n]); // [[1,-1],[2,-2],[3,-3]] [[1,-1],[2,-2],[3,-3]].flat(); // [1,-1,2,-2,3,-3] [1,2,3].flatMap(n => [n,-n]); // [1,-1,2,-2,3,-3]
flatMapを使って1次元にする
コード
flatMapを使って、どんな深さの配列でも1次元に落とし込みます。これは「無限回flat()を行い、1次元になった時点でやめる」のと等価なので、flatInfと名付けました。
const arr = [1, [2, [3, 4, [5]]]]; const flatInf = item => { if (Array.isArray(item)) return item.flatMap(flatInf); return item; }; console.log(flatInf(arr)); // [1, 2, 3, 4, 5]
flatInfの動作
- 配列の各要素を「1次元の配列あるいは要素」にする
- できた配列をflat()する
これだけです。配列の要素が1次元の配列か要素のみなら、flat()するだけで1次元になります。
// 1次元の配列か要素のみの配列 [[1, 2], 3, [4, 5, 6], 7, 8, 9].flat(); // [1, 2, 3, 4, 5, 6, 7, 8, 9]
与えられる配列の各要素の深さは1次元とは限りません。コレを1次元にするために、再帰的にflatInfを適用しています。
おまけ: コードを圧縮する
そこまで変わりませんが。
const arr = [1, [2, [3, 4, [5]]]]; const flatInf=a=>Array.isArray(a)?a.flatMap(flatInf):a; console.log(flatInf(arr)); // [1, 2, 3, 4, 5]
