[Ruby] 配列内のある複数の要素が一致したらその要素を取得したい

Question

###環境
ruby 2.3.1p112 (2016-04-26 revision 54768) [x86_64-darwin17]

###実現したいこと

arr1 = [["a", "b", "c"],["1", "2", "3"]]
arr2 = [["a", "b", "x"],["4", "5", "6"]]

上記のように、arr1とarr2の最初の配列に格納されている、初めから2つの要素"a"、"b"が等しい場合、その要素を取得する、といったことがしたいと思っています。

そのようなメソッドが見つけられないのですが、なにか方法はあるでしょうか？

ただ、以下のようにeach文の中にeach文を記述する以外の方法でお願いいたします。

arr3 = []
arr1.each do |a1|
  arr2.each do |a2|
    if a1[0] == a2[0] && a1[1] == a2[1]
      arr3.push(a1)
    end
  end
end

何かご教示いただけることがあれば、何卒よろしくお願いいたします。

Comments

[maisumakun]

arr1とarr2を比較するのは、「同じ位置」のものだけですか？それとも、二重eachのコードの実行結果のように「位置は無視して」比較する感じでしょうか？

[s_diff]

ご質問ありがとうございます。
今回お聞きしたいのは「同じ位置」のものだけですが、できれば両方お聞きしたいです。
よろしくお願いいたします。

Answer 1

位置を無視した場合は

arr1 = [%w[a b c], %w[1 2 3], %w[a b d]]
arr2 = [%w[a b x], %w[4 5 6], %w[1 0 3]]

hash2 = arr2.map{|x,y,z| [[x,y], true]}.to_h
arr3 = arr1.select{|x,y,z| hash2.key?([x,y]) }

同じ位置のみなら

arr1 = [%w[a b c], %w[1 2 3], %w[a b d]]
arr2 = [%w[a b x], %w[4 5 6], %w[1 0 3]]

arr3 = arr1.select.with_index do |(x,y,z),ix|
         a,b,c = arr2[ix]
         a == x && b == y
       end

Answer 2

方法はいろいろ考えられます。

# 同じ位置だけ見比べる
arr1.zip(arr2).select{ |a, b| a[0..1] == b[0..1] }.map(&:first)

# 全部比較する
hash1 = arr1.group_by{ |arr| arr[0..1] }
hash2 = arr2.group_by{ |arr| arr[0..1] }
hash1.values_at(*hash1.keys & hash2.keys).flatten(1)

Comments

[s_diff]

そのような便利なメソッドがあるのですね。
勉強になりました。ありがとうございます！