Combinación de dos matrices en Ruby

Question

¿Cuál es la forma de Ruby para lograr lo siguiente?

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

Quiero una matriz:

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

Answer 1

Puede utilizar product para obtener el producto cartesiano de los conjuntos primero, se recogen los resultados de la función.

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

Así que usted puede utilizar map o collect para obtener los resultados. Son nombres diferentes para el mismo método.

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

Answer 2

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

Nota: En 1.8.7+ puede agregar 1 como argumento para aplanar, por lo que aún obtendrá resultados correctos cuando vuelve f una matriz.

Aquí es una abstracción para un número arbitrario de matrices:

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] 

Answer 3

Facetas tiene Array#product que le dará el producto vectorial de las matrices. También se alias como ** operator para el caso de dos matrices. Usando esa, se vería así:

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

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

Si está usando Ruby 1.9, product es una función incorporada Array.