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

82

Каким способом Ruby добиться подписания?

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

Я хочу массив:

=> [f(1,3) ,f(1,4) , f(2,3) ,f(2,4)]
Pierrotlefou
источник
@lucasarruda, это не совсем то, что просит OP. Zip создаст массив из файлов [a[0], b[0]]и [a[1], b[1]]. Он не будет включать [a[0], b[1]]. a.zip(b) => [[1,3],[2,4]]
KNejad
Спасибо @KNejad, вы правы. Удалил мой комментарий.
lucasarruda

Ответы:

149

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

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

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

a.product(b).collect { |x, y| f(x, y) }
Аарон Хинни
источник
Продукт встроен только в версии 1.9.
Песто,
7
Продукт находится в версии 1.8.7 - я использую его, чтобы делать именно то, о чем просит Пьер
Крис МакКоли,
1
@FloatingRock неверно, это комбинация, а не пермутация. См. Ruby-doc.org/core-2.1.3/Array.html#method-i-product
Джош Дил
1
@FloatingRock Не так. Первый элемент каждого элемента всегда будет происходить из первого списка, второй элемент из второго списка и так далее. Это все возможные комбинации элементов, но они никогда не меняются, так что это не перестановки.
Марк Рид
1
Для трех массивов a.product(b,c)и т.д. (пока не закончится память). Руби такой классный!
Обромиос
11
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]
sepp2k
источник
Это сработает, хотя я не уверен, что это лучший способ (хотя и не могу придумать лучшего, так что +1)
Роберт Массайоли
Хотел бы я +2 за комбинацию произвольного количества массивов. Очень круто!
pierrotlefou,
Я тоже выступил с этим ... но потом я знал, что это должен быть более простой способ сделать это ... +1 за это, но @Aaron Hinni отвечает своему уборщику
Орландо
5

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 # - это 1.9изм? По крайней мере, это в 1.8.7.
Марк Рид,