Mapeando valores de dois arrays em Ruby

Question

Gostaria de saber se existe uma maneira de fazer o que posso fazer abaixo com Python, em Ruby:

sum = reduce(lambda x, y: x + y, map(lambda x, y: x * y, weights, data))

Eu tenho dois arrays de tamanhos iguais com os pesos e dados, mas não consigo encontrar uma função semelhante ao map em Ruby, reduzo que estou funcionando.

Answer 1

@Michiel de Mare

Seu exemplo do Ruby 1.9 pode ser um pouco mais reduzido:

weights.zip(data).map(:*).reduce(:+)

Observe também que no Ruby 1.8, se você precisar do ActiveSupport (do Rails), você pode usar:

weights.zip(data).map(&:*).reduce(&:+)

Answer 2

Em Ruby 1.9:

weights.zip(data).map{|a,b| a*b}.reduce(:+)

Em Ruby 1.8:

weights.zip(data).inject(0) {|sum,(w,d)| sum + w*d }

Answer 3

A função Array.zip faz uma combinação elemento a elemento de arrays.Não é tão limpo quanto a sintaxe do Python, mas aqui está uma abordagem que você pode usar:

weights = [1, 2, 3]
data = [4, 5, 6]
result = Array.new
a.zip(b) { |x, y| result << x * y } # For just the one operation

sum = 0
a.zip(b) { |x, y| sum += x * y } # For both operations

Answer 4

Rubi tem um map método (também conhecido comoo collect método), que pode ser aplicado a qualquer Enumerable objeto.Se numbers é uma matriz de números, a seguinte linha em Ruby:

numbers.map{|x| x + 5}

é o equivalente à seguinte linha em Python:

map(lambda x: x + 5, numbers)

Para mais detalhes, consulte aqui ou aqui.

Answer 5

Uma alternativa para o mapa que também funciona para mais de 2 arrays:

def dot(*arrays)
  arrays.transpose.map {|vals| yield vals}
end

dot(weights,data) {|a,b| a*b} 

# OR, if you have a third array

dot(weights,data,offsets) {|a,b,c| (a*b)+c}

Isso também pode ser adicionado ao Array:

class Array
  def dot
    self.transpose.map{|vals| yield vals}
  end
end

[weights,data].dot {|a,b| a*b}

#OR

[weights,data,offsets].dot {|a,b,c| (a*b)+c}

Answer 6

weights = [1,2,3]
data    = [10,50,30]

require 'matrix'
Vector[*weights].inner_product Vector[*data] # => 200