ruby - 使用 Ruby 按位置添加二维数组中所有元素的最清晰方法是什么？

Question

在 Ruby 中按位置添加二维数组中所有元素的最清晰和最有效的方法是什么。示例:

2darray = [[1,2,3],[1,2,3]]
result = [2,4,6]

我有以下代码

def sum_elements_by_position(array)
 total_elements = array.length
 result = []

 for i in 0...array.first.length
   n = 0
   array.each { |subarray| n += subarray[i] }
   result << n
 end
 result
end

假设:所有主要元素的长度都相同

对于奖励积分，很高兴看到一个解决方案适用于任意长度的主要元素

Answer 1

您可以压缩第一行和其余行，然后求和:

def sum_elements_by_position(array)
    array[0].zip(*array[1..-1]).map do |col|
        col.inject(:+)
    end
end