web-dev-qa-db-ja.com

Rubyの2つの配列の組み合わせ

Rubyフォローを達成する方法は何ですか？

a = [1,2]
b = [3,4]

配列が欲しい：

=> [f(1,3) ,f(1,4) , f(2,3) ,f(2,4)]
rubycombinations
70
pierrotlefou

productを使用して、最初に配列のデカルト積を取得してから、関数の結果を収集できます。

a.product(b) => [[1, 3], [1, 4], [2, 3], [2, 4]]

したがって、mapまたはcollectを使用して結果を取得できます。同じメソッドの異なる名前です。

a.product(b).collect { |x, y| f(x, y) }
133
Aaron Hinni
a.map {|x| b.map {|y| f(x,y) } }.flatten

注：1.8.7以降では、1をフラット化の引数として使用するため、fが配列を返す場合でも正しい結果が得られます。

任意の数の配列の抽象化を次に示します。

def combine_arrays(*arrays)
  if arrays.empty?
    yield
  else
    first, *rest = arrays
    first.map do |x|
      combine_arrays(*rest) {|*args| yield x, *args }
    end.flatten
      #.flatten(1)
  end
end

combine_arrays([1,2,3],[3,4,5],[6,7,8]) do |x,y,z| x+y+z end
# => [10, 11, 12, 11, 12, 13, 12, 13, 14, 11, 12, 13, 12, 13, 14, 13, 14, 15, 12, 13, 14, 13, 14, 15, 14, 15, 16]
11
sepp2k

FacetsにはArray#productがあり、配列のクロス積を提供します。また、2配列の場合は** operatorとしてエイリアスされます。それを使用すると、次のようになります。

require 'facets/array'
a = [1,2]
b = [3,4]

(a.product b).collect {|x, y| f(x, y)}

Ruby 1.9）を使用している場合、 product は組み込みの配列関数です。

5
Pesto