ruby - 在 Ruby 中查找二维数组的总和

Question

我有一个数组二维数组。我想创建一个新的二维数组，它在二维数组中找到这些值的总和。

新数组在 x,y 处的总和 = arr1 在 x,y 处的总和 + arr2 在 x,y 处的总和 + ....

|1,2,4| |1,1,1| |1,1,1| 
|2,4,6| |1,1,1| |1,1,1| 
|2,4,6| |1,1,1| |1,1,1| 
|2,4,6| |1,1,1| |1,1,1| 

现在将上面的二维数组相加会得到，

|3,4,6|
|4,6,8|
|4,6,8|
|4,6,8|

如何在 Ruby 中实现这一点(而不是在任何其他语言中)。我写了一个方法，但是看起来又长又难看。

Answer 1

require 'matrix'

Matrix[
  [1, 2, 4],
  [2, 4, 6],
  [2, 4, 6],
  [2, 4, 6]
] +

Matrix[
  [1, 1, 1],
  [1, 1, 1],
  [1, 1, 1],
  [1, 1, 1]
] +

Matrix[
  [1, 1, 1],
  [1, 1, 1],
  [1, 1, 1],
  [1, 1, 1]
]

# => Matrix[[3, 4, 6], [4, 6, 8], [4, 6, 8], [4, 6, 8]]

或者，如果您想要与@Jeriko 的回答相同的格式，即返回一个 Array 而不是 Matrix:

def sum_arrays(*a)
  return *a.map {|m| Matrix[*m] }.reduce(:+)
end

# data you supplied:
x = [[1, 2, 4], [2, 4, 6], [2, 4, 6], [2, 4, 6]]
y = [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]]
z = [[1, 1, 1], [1, 1, 1], [1, 1, 1], [1, 1, 1]]

p sum_arrays(x, y, z)
# => [[3, 4, 6], [4, 6, 8], [4, 6, 8], [4, 6, 8]]