2009-10-09 6 views
59

Quel est le moyen de Ruby à suivre?Combinaison de deux tableaux en Ruby

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

Je veux un tableau:

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

Ce serait la bonne réponse maintenant http://stackoverflow.com/a/12011316/135767 même si http://stackoverflow.com/questions/12011294/combining-two-arrays-to-create-a-two -dimensional-array-in-ruby est une copie de cette question. – lucasarruda

Répondre

116

Vous pouvez utiliser product pour obtenir le produit cartésien des tableaux, puis recueillir les résultats de la fonction.

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

Vous pouvez utiliser map ou collect pour obtenir les résultats. Ce sont des noms différents pour la même méthode.

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

Le produit est seulement intégré à partir de 1.9. – Pesto

+7

Le produit est en 1.8.7 - Je l'utilise pour faire exactement ce que Pierr demande –

+1

@FloatingRock c'est incorrect c'est combinaison non permuation. Voir http://www.ruby-doc.org/core-2.1.3/Array.html#method-i-product –

8
a.map {|x| b.map {|y| f(x,y) } }.flatten 

Remarque: vous pouvez ajouter 1.8.7+ 1 comme argument pour aplatir, de sorte que vous aurez toujours obtenir des résultats corrects lors f retours un tableau.

est ici une abstraction pour un nombre arbitraire de tableaux:

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] 
+0

Cela fonctionnera bien que je ne sache pas si c'est la meilleure façon (mais ne peut pas penser à un meilleur donc +1) –

+0

Je souhaite que je pourrais +2 pour la combinaison de quantité arbitraire de tableaux. Très sympa! – pierrotlefou

+0

je suis sorti avec cela aussi .. mais alors je savais que theres devrait être un moyen plus facile de le faire ... +1 pour cela, mais le @Aaron Hinni répondre à son nettoyeur – Orlando

4

Facettes a Array#product qui vous donnera le produit croisé de tableaux. Il est également aliasé ** operator pour le cas de deux matrices. En utilisant cela, il ressemblerait à ceci:

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

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

Si vous utilisez Ruby 1.9, product est fonction Array intégré.

+0

Pourquoi les gens continuent de dire Array # product est un 1.9ism? C'est en 1.8.7, au moins. –

+3

http://apidock.com/ruby/Array/product: c'était en 1.8.7-p72. –