配列内の特定の要素が未定義なら初期化してからpush、 定義済みならそのままpushという操作をスマートに書く方法

Question

配列の特定の要素が未定義なら初期化してからpush、
定義済みならそのままpushという操作をスマートに書く方法を考えています。
例えば以下のような場合です。

const array = [];

array[0] = ['0'];
array[1] = ['1'];
array[3] = ['3'];

for (let i = 0; i <= 5; i++) {
  if (array[i] === undefined) array[i] = [];
  array[i].push('add');
}

console.table(array);

// while文実行後のarray配列
// array = [
//   ['0', 'add'],
//   ['1', 'add'],
//   ['add'],
//   ['3', 'add'],
//   ['add'],
//   ['add'],
// ];

for文の中でarray[i]が3回もあって何だか冗長に見えます。
もっと簡潔な記述がありましたらご教授の程よろしくお願いします。

Answer 1

for文の中でarray[i]が3回もあって何だか冗長に見えます。

個人的には、何をしているのかわかりやすくて悪くないと思いますけれども……

一応、スマート（）に書いてみましたが、可読性はtoromouさんのコードの方が優れていると思います。

for (let i = 0; i <= 5; i++) {
  ( array[i] || ( array[i] = [] ) ).push('add');
}

###質問クローズを受けて追記

jun68yktさんなどが挙げたスプレッド演算子を使って代入する方法は、破壊的なので、元のコードと微妙に振る舞いが違います。
あまり変わったことをしていなければ問題ないとは思いますが、お気を付けください。

const array = [];

array[0] = ['0'];
array[1] = ['1'];
array[3] = ['3'];
const old = array[3];

for (let i = 0; i <= 5; i++) {
  array[i] = [...(array[i] || []), 'add'];
}

console.table(old === array[3]); // false

const array_ = [];

array_[0] = ['0'];
array_[1] = ['1'];
array_[3] = ['3'];
const old_ = array_[3];

for (let i = 0; i <= 5; i++) {
  if (array_[i] === undefined) array_[i] = [];
  array_[i].push('add');
}

console.table(old_ === array_[3]); // true

Comments

[toromou]

確かにそうですね…完全に見落としていました。
Lhankor_Mhyさんやkei344さんから挙げていただいた方法は非破壊的なのでそちらの方が安全かもしれませんね。
ご親切にありがとうございました。参考にさせていただきます。

Answer 2

こんにちは。スプレッド構文を使ってみるのはいかがでしょうか？

  array[i] = [...(array[i] || []), 'add'];

• 動作確認用ＣｏｄｅＰｅｎ： https://codepen.io/jun68ykt/pen/LYVpaOm?editors=0012

Comments

[toromou]

なるほど。"||"を使えば簡潔に書けそうだとは思っていましたが、スプレッド構文と組み合わせて使うというのは目から鱗でした。
ただ、他の方のご回答にもあるように可読性を考えると元のままでもいい気もしてきました。

一番早くお答えいただいたのでこちらをベストアンサーとさせて頂きます。
ご回答いただいた皆様ありがとうございました。

[jun68ykt]

どういたしまして。参考になれば幸いです。

Answer 3

undefinedに対して追加することが確定しているなら。

// ( array[i] || [] ).push( 'add' );
// 代入していなかった･･･。
   ( array[i] || ( array[i] = [] ) ).push('add');

Comments

[kei344]

（あ、まちがえた）

Answer 4

  array[i] = [...array[i] || [], 'add'];