Сопоставление значений из двух массивов в Ruby
Вопрос
Мне интересно, есть ли способ сделать то, что я могу сделать ниже, с Python, в Ruby:
sum = reduce(lambda x, y: x + y, map(lambda x, y: x * y, weights, data))
У меня есть два массива равных размеров с весами и данными, но, похоже, я не могу найти функцию, похожую на map в Ruby, reduce у меня работает.
Решение
@Michiel de Mare
Ваш пример Ruby 1.9 можно еще немного сократить:
weights.zip(data).map(:*).reduce(:+)
Также обратите внимание, что в Ruby 1.8, если вам требуется ActiveSupport (из Rails), вы можете использовать:
weights.zip(data).map(&:*).reduce(&:+)
Другие советы
В 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 }
Функция 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
У Руби есть map
метод (он жетот самый collect
метод), который может быть применен к любому Enumerable
объект.Если numbers
представляет собой массив чисел, следующая строка в Ruby:
numbers.map{|x| x + 5}
является эквивалентом следующей строки в Python:
map(lambda x: x + 5, numbers)
Для получения более подробной информации см. здесь или здесь.
Альтернатива для карты, которая также работает для более чем 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}
Это также может быть добавлено в 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}
weights = [1,2,3]
data = [10,50,30]
require 'matrix'
Vector[*weights].inner_product Vector[*data] # => 200