Комбинация двух массивов в Ruby

Ответ 1

Вы можете использовать product, чтобы сначала получить декартово произведение массивов, а затем собрать результаты функции.

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

Таким образом, вы можете использовать map или collect для получения результатов. Это разные имена для одного и того же метода.

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

Ответ 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]

Ответ 3

Facets имеет 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 - это встроенная функция Array.