【rails】二重配列からの値の取得

Question

###以下の様な二重配列から値を取得するにはどうすれば良いでしょうか？

配列[ [1, nil], [5, 123], [3, 1900], [4, nil] ]をインスタンス変数＠arrayとする。

①二重配列の下層配列、添字0の数字と合致するかどうか
②合致した場合、その下層配列の添字１の数値を取得

例）上記の配列を対象とした場合、5で＠arrayを検索し、「5」があるので、「123」を取得するといった具合です。

以下の様な検索の仕方は思いついたものの、HITした場合に、その配列の添字１の値を取得する方法まで思い浮かびません・・・。

＠array[0].include?(5)

Answer 1

複数ヒットした場合、一つもヒットしなかった場合のことを考慮し、 配列を返すことにしてみました。

a.rb

def my_filter(ary, v)
  ary.inject([]) do |memo, vec|
    memo << vec[1] if vec[0] == v
    memo
  end
end

def my_filter2(ary, v)
  ary.select { |vec| vec[0] == v }.map { |vec| vec[1] }
end

p my_filter([ [1, nil], [5, 123], [3, 1900], [4, nil] ], 5)
p my_filter([ [5, 123], [5, 1900] ], 5)
p my_filter([ [1, 123] ], 5)
p my_filter([ [] ], 5)
p ''

p my_filter2([ [1, nil], [5, 123], [3, 1900], [4, nil] ], 5)
p my_filter2([ [5, 123], [5, 1900] ], 5)
p my_filter2([ [1, 123] ], 5)
p my_filter2([ [] ], 5)

実行結果

$ ruby a.rb
[123]
[123, 1900]
[]
[]
""
[123]
[123, 1900]
[]
[]

参考

• mapなどの配列の値を操作するメソッドまとめ http://qiita.com/shibukk/items/d985d014de598d925b8b

Answer 2

findメソッドで要素を取得して、添字1を取得する方法が使えると思います。

例えば添字0が5の要素の添字1を取得するときには、findで[0]が5の要素を取得して、[1]で123を取得することができます。

@array = [ [1, nil], [5, 123], [3, 1900], [4, nil] ]

@array.find{|arr| arr[0] == 5 }[1]
#=> 123

Comments

[gogoackman3]

いつもありがとうございます！実装したところ、無事いけましたー！
この書き方は思いつきませんでした、助かりました！！