Combinación de dos matrices en Ruby


82

¿Cuál es la forma Ruby de conseguir seguidores?

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

Quiero una matriz:

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

@lucasarruda eso no es exactamente lo mismo que pide OP. Zip creará una matriz de [a[0], b[0]]y [a[1], b[1]]. No incluiría [a[0], b[1]]. a.zip(b) => [[1,3],[2,4]]
KNejad

Gracias @KNejad, tienes razón. Borré mi comentario.
lucasarruda

Respuestas:


147

Puede usar productpara obtener el producto cartesiano de las matrices primero y luego recopilar los resultados de la función.

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

Entonces puedes usar mapo collectpara obtener los resultados. Son nombres diferentes para el mismo método.

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

El producto solo está integrado a partir de 1.9.
Pesto

7
El producto está en 1.8.7 - Lo uso para hacer exactamente lo que pide Pierr
Chris McCauley

1
@FloatingRock eso es incorrecto, es una combinación, no una permutación. Ver ruby-doc.org/core-2.1.3/Array.html#method-i-product
Josh Diehl

1
@FloatingRock No es así. El primer elemento de cada elemento siempre vendrá de la primera lista, el segundo elemento de la segunda lista, y así sucesivamente. Son todas las combinaciones posibles de elementos, pero nunca se reordenan, por lo que no son permutaciones.
Mark Reed

Y si tienes c = [5,6] yd = [7,8] puedes hacerlo[a,b,c,d].reduce([[]]){|r,e| r=r.product(e)}.map{|x| x.flatten}
xxjjnn

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

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

Aquí hay 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]

Esto funcionará, aunque no estoy seguro de si es la mejor manera (aunque no puedo pensar en una mejor, así que +1)
Robert Massaioli

Ojalá pudiera +2 por la combinación de una cantidad arbitraria de matrices. ¡Muy genial!
pierrotlefou

También salí con esto ... pero luego supe que debería haber una manera más fácil de hacerlo ... +1 para esto, pero @Aaron Hinni responde que es más limpio
Orlando

5

Facetas tiene lo Array#productque le dará el producto cruzado de matrices. También tiene el alias ** operatorpara el caso de dos matrices. Usando eso, se vería así:

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

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

Si está utilizando Ruby 1.9, productes una función de matriz incorporada.


¿Por qué la gente sigue diciendo que el producto Array # es un 1.9ism? Está en 1.8.7, al menos.
Mark Reed
Al usar nuestro sitio, usted reconoce que ha leído y comprende nuestra Política de Cookies y Política de Privacidad.
Licensed under cc by-sa 3.0 with attribution required.