0
Отображение значений из двух массивов в Ruby
Мне интересно, есть ли способ сделать то, что я могу сделать с Python, в Ruby:
sum = reduce(lambda x, y: x + y, map(lambda x, y: x * y, weights, data))
У меня есть два массива равных размеров с весами и данными, но я не могу найти функцию, похожую на map в Ruby, уменьшить, у меня есть работа.
Ответов (6)6
Решение
0
Функция 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
0
Альтернатива карте, которая также работает для более чем двух массивов:
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}