Combinaison de deux tableaux en Ruby
-
20-09-2019 - |
Question
Quelle est la façon Ruby pour atteindre suivant?
a = [1,2]
b = [3,4]
Je veux un tableau:
=> [f(1,3) ,f(1,4) , f(2,3) ,f(2,4)]
La solution
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. Ils sont des noms différents pour la même méthode.
a.product(b).collect { |x, y| f(x, y) }
Autres conseils
a.map {|x| b.map {|y| f(x,y) } }.flatten
Note:. Sur 1.8.7+, vous pouvez ajouter 1
comme argument pour aplatir, de sorte que vous aurez toujours obtenir des résultats corrects quand f
retourne un tableau
Voici 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]
Facettes a Array#product
qui vous donnera le produit croisé de tableaux. Il est également Lissé comme le ** operator
pour le cas de deux tableau. 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 une fonction de matrice intégrée.