Połączenie dwóch tablic w Rubim

82

Jaki jest sposób Ruby na osiągnięcie obserwowania?

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

Chcę tablicę:

=> [f(1,3) ,f(1,4) , f(2,3) ,f(2,4)]
pierrotlefou
źródło
@lucasarruda to nie to samo, o co pyta OP. Zip utworzy tablicę [a[0], b[0]]i [a[1], b[1]]. Nie obejmowałoby [a[0], b[1]]. a.zip(b) => [[1,3],[2,4]]
KNejad
Dzięki @KNejad, masz rację. Usunąłem mój komentarz.
lucasarruda

Odpowiedzi:

148

Możesz użyć, productaby najpierw uzyskać iloczyn kartezjański tablic, a następnie zebrać wyniki funkcji.

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

Możesz więc użyć maplub, collectaby uzyskać wyniki. Są to różne nazwy tej samej metody.

a.product(b).collect { |x, y| f(x, y) }
Aaron Hinni
źródło
Produkt jest wbudowany dopiero od 1.9.
Pesto
7
Produkt jest w wersji 1.8.7 - używam go, aby zrobić dokładnie to, o co prosi Pierr
Chris McCauley
1
@FloatingRock, który jest nieprawidłowy, to kombinacja, a nie permuacja. Zobacz ruby-doc.org/core-2.1.3/Array.html#method-i-product
Josh Diehl
1
@FloatingRock Nie tak. Pierwszy element każdej pozycji zawsze będzie pochodził z pierwszej listy, drugi element z drugiej listy i tak dalej. To wszystkie możliwe kombinacje elementów, ale nigdy nie są zmieniane, więc nie są to permutacje.
Mark Reed
1
Przez trzy tablice a.product(b,c)i tak dalej (do wyczerpania pamięci). Ruby jest super!
Obromios
11
a.map {|x| b.map {|y| f(x,y) } }.flatten

Uwaga: W wersji 1.8.7+ możesz dodać 1jako argument spłaszczenia, aby nadal uzyskiwać poprawne wyniki po fzwróceniu tablicy.

Oto abstrakcja dla dowolnej liczby tablic:

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
źródło
To zadziała, chociaż nie jestem pewien, czy to najlepszy sposób (chociaż nie mogę wymyślić lepszego, więc +1)
Robert Massaioli
Chciałbym móc +2 za kombinację dowolnej liczby tablic. Bardzo fajny!
pierrotlefou
ja też z tym wyszedłem ... ale wiedziałem, że powinien być łatwiejszy sposób na zrobienie tego ... +1 za to, ale @Aaron Hinni odpowiada na jego czystsze
Orlando
5

Aspekty mają, Array#productktóre dają iloczyn krzyżowy tablic. Jest również aliasowany jak ** operatordla przypadku dwóch tablic. Używając tego, wyglądałoby to tak:

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

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

Jeśli używasz Ruby 1.9, productjest wbudowana funkcja Array.

pesto
źródło
Dlaczego ludzie ciągle mówią, że produkt Array # to 1,9ism? Jest przynajmniej w wersji 1.8.7.
Mark Reed