Сопоставление значений из двух массивов в Ruby

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

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

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


person Tim Trueman    schedule 06.08.2008    source источник


Ответы (5)


@Мишель де Маре

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

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

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

weights.zip(data).map(&:*).reduce(&:+)
person Lily Ballard    schedule 07.08.2008
comment
У меня это не работает на Ruby 1.9.3 (ArgumentError): [1,2,3].zip([4,5,6]).map(:*) - person Felix Rabe; 12.05.2012
comment
@FelixRabe: кажется, ruby ​​изменил способ обработки аргументов в такого рода процедурах. Я действительно не знаю никаких подробностей, так как не прикасался к рубину несколько лет. - person Lily Ballard; 12.05.2012

В Руби 1.9:

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

В Руби 1.8:

weights.zip(data).inject(0) {|sum,(w,d)| sum + w*d }
person Michiel de Mare    schedule 07.08.2008

Функция 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
person Curt Hagenlocher    schedule 06.08.2008

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

numbers.map{|x| x + 5}

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

map(lambda x: x + 5, numbers)

Для получения дополнительной информации см. здесь или здесь.

person Joey deVilla    schedule 06.08.2008

Альтернатива для карты, которая работает и для более чем 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}
person Chris Turner    schedule 26.02.2011