在Ruby中映射两个数组中的值

时间:2020-03-05 18:38:00  来源:igfitidea点击:

我想知道是否有一种方法可以在Ruby中使用Python进行下面的工作:

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

我有两个大小和权重相等的数组,但是我似乎找不到与Ruby中的map类似的函数,请减少工作量。

解决方案

回答

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

回答

Ruby有一个map方法(又称collect方法),可以应用于任何Enumerable对象。如果`numbers'是数字数组,则Ruby中的以下行:

numbers.map{|x| x + 5}

与Python中的以下行等效:

map(lambda x: x + 5, numbers)

有关更多详细信息,请参见此处或者此处。

回答

在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 }

回答

@Michiel de Mare

Ruby 1.9示例可以进一步简化:

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

还要注意,在Ruby 1.8中,如果需要ActiveSupport(来自Rails),则可以使用:

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