2008-08-06 11 views
18

мне интересно, если есть способ сделать то, что я могу сделать ниже с Python, в Ruby:значение Mapping из два массива в Рубите

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

У меня есть два массива одинакового размера с весами и данные но я не могу найти функцию, похожую на карту в Ruby, уменьшая работу.

ответ

13

@Michiel-де-Маре

Ваш пример Ruby 1.9 может быть сокращен немного дальше:

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

Также обратите внимание, что в Рубин 1.8, если вам требуется ActiveSupport (от Rails) вы можете использовать:

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

Это не работает для меня на Руби 1.9.3 (ArgumentError). `[1,2,3] .zip ([4,5,6]) отображение (: *) ` – 2012-05-11 23:47:24

1

Функция Array.zip делает элементную комбинацию массивов. Это не совсем так чисто, как синтаксис Python, но вот один подход можно использовать:

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 
1

Рубин имеет map метод (также называется методом collect), который может быть применен к любому Enumerable объекта. Если numbers массив чисел, следующая строка в Ruby:

numbers.map{|x| x + 5} 

является эквивалентом следующей строки в Python:

map(lambda x: x + 5, numbers) 

Для получения более подробной информации см here или here.

5

В Ruby 1.9:

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

В Ruby 1.8:

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

Альтернатива для карты, которая работает более 2-х массивов, а также:

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} 

Это также может быть добавлен Массив:

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} 
0
weights = [1,2,3] 
data = [10,50,30] 

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