在红宝石两个阵列的组合(Combination of two arrays in Ruby)

2019-07-21 17:03发布

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

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

我想一个数组:

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

Answer 1:

您可以使用product先拿到阵列的笛卡尔积,然后收集功能的结果。

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

所以,你可以使用mapcollect得到的结果。 他们是同样的方法不同的名字。

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


Answer 2:

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]


Answer 3:

面有Array#product ,这将给你的阵列的交叉乘积。 它也别名为** operator对于两个阵列的情况。 利用这一点,它应该是这样的:

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

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

如果您在使用Ruby 1.9, product是内置阵列功能。



文章来源: Combination of two arrays in Ruby