Ruby ２次元配列 要素 挿入

Question

Ruby初心者です。

Rubyの２次元配列にpushをしようとしています。例えば

[ [0, 0, 0],
[0, 0, 0] ]

が用意されている２次元配列に新たにpushをして

[ [0, 0, 0, 1, 1],
[0, 0, 0] ]

ができるかどうか、または

[ [0, 0, 0, 1, 1],
[0, 0, 0],
[0, 0, 0]]

ができるかどうか教えていただきたいです。

やりたいことはテキストファイルに

1
内容
2
内容
内容
3
内容
.
.
.
```というファイルがあってこれを
[[1, 内容],
 [2,内容,内容],
 [3,内容]]
という配列にしたいと思っています。

現在書いているのはこのコードです。
```Ruby
#! ruby -Ku

cur_dir = Dir.pwd

before_list = []
after_list = Array.new(2) { Array.new(2,0) }
list_size = 0

File.open(cur_dir + "/hoge.txt", "r") do |file|
  before_list = file.readlines
end

before_list.each { |line|
  contents_size = 0
  if line.match(/^[0-9]+$/)
    after_list[list_size][0] = line[0]
    list_size += 1
  else
    after_list[list_size].push(hoge)
    contents_size += 1
  end
}

をしても
undefined method 'push' for nil:NilClass(NoMethodError)
と出てきてしまいます。

Comments

[mather]

hoge, hogehoge は具体的になんですか？

[kr_]

編集しました。

[mather]

編集というか、聞きたいことが全然違いましたね…。
質問としてはエラーの原因と対処法を知りたいということで良いですか？
そもそも「列挿入」という言い方は紛らわしいのでやめましょう。

[kr_]

すみません…
はい、原因と対処法を知りたいです。
申し訳ありません。

Answer 1

エラーの原因について、処理を順番に追っていくとわかるのですが、
途中で list_size == 2 になります。(テキストファイル内の 2 を処理した直後)
after_list のサイズは 2 なので、 after_list[2] は要素が存在せず nil を返します。
このとき "内容" という文字列を挿入しようとするため nil.push(hoge) となってしまいエラーが発せします。
（そもそも hoge はどこにも宣言されていない変数なので意味がわかりませんが…）

対処方法ですが、以下のようになるかと思います。実装は書かないので試してみてください。

• after_list は空配列でスタート
• 数字を見つけたら after_list に数字の要素をもつ配列 [line[0]] を追加する
• 数字以外だったら after_list の「最後の要素」にあたる配列に、要素を追加する

Comments

[kr_]

お手数おかけしてしまい申し訳ありませんでした。ありがとうございました。

Answer 2

elements = [["UCLA","UCB"],["UCSD","UCSF","MIT"]]
elements[0].push("Harvard")
p elements[0]
=> ["UCLA", "UCB", "Harvard"]

値の追加例です。