Ruby中两个数组的组合

什么是Ruby的方式来实现以下?

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

我想要一个数组:

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

您可以先使用product获得数组的笛卡尔积,然后收集函数结果。

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

所以你可以使用mapcollect来获得结果。 对于相同的方法,它们是不同的名称。

 a.product(b).collect { |x, y| f(x, y) } 
 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] 

Facets拥有Array#product ,它将为您提供数组的交叉产品。 它也被视为双arrays情况下的** operator 。 使用它,它会看起来像这样:

 require 'facets/array' a = [1,2] b = [3,4] (a.product b).collect {|x, y| f(x, y)} 

如果您使用的是Ruby 1.9,则product是一个内置的Array函数。