从Ruby中的两个数组映射值

Posted 2023-03-16

【中文标题】从Ruby中的两个数组映射值

【英文标题】：Mapping values from two array in Ruby

【发布时间】：2010-09-05 10:05:41

【问题描述】：

我想知道是否有办法在 Ruby 中使用 Python 完成以下操作：

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

我有两个大小相等的数组，权重和数据，但我似乎无法在 Ruby 中找到类似于 map 的函数，reduce 我正在工作。

【参考方案1】：

@Michiel de Mare

您的 Ruby 1.9 示例可以进一步缩短：

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

另请注意，在 Ruby 1.8 中，如果您需要 ActiveSupport（来自 Rails），您可以使用：

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

【讨论】：

这对我在 Ruby 1.9.3 (ArgumentError) 上不起作用：[1,2,3].zip([4,5,6]).map(:*)

@FelixRabe：似乎 ruby​​ 改变了在这种 proc 中处理参数的方式。我真的不知道任何细节，因为我已经有几年没有接触过红宝石了。

【参考方案2】：

在 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 

【参考方案3】：

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

【参考方案4】：

Ruby 有一个map 方法（又名collect 方法），它可以应用于任何Enumerable 对象。如果numbers 是一个数字数组，Ruby 中的以下行：

numbers.map|x| x + 5

相当于 Python 中的以下行：

map(lambda x: x + 5, numbers)

更多详情，请参阅here 或here。

【参考方案5】：

另一种适用于超过 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

【参考方案6】：

weights = [1,2,3]
data    = [10,50,30]

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