Ruby-添加/減去一個數組元素與另一個數組

Question

我這樣做：

a = [1,2,3,4] 
b = [2,3,4,5] 
c = b - a 
put c 

我得到這個答案 - >[1]
我想這個答案 - >[1,1,1,1]（如矩陣加法/減法）

我嘗試這樣做：

c.each {|e| c[e] = b[e] - a[e]} 

，但我得到這樣的回答：[1,0,0,0]

有人可以給我一個正確的方法來做到這一點嗎？非常感謝！

Answer 1

你可以使用zip：

a.zip(b).map { |x, y| y - x } 
# => [1, 1, 1, 1] 

還有一類Matrix：

require "matrix" 

a = Matrix[[1, 2, 3, 4]] 
b = Matrix[[2, 3, 4, 5]] 
c = b - a 
# => Matrix[[1, 1, 1, 1]] 

Answer 2

您可以使用each_with_index和map。

c = b.each_with_index.map { |n,i| n - a[i] } 
# => [1, 1, 1, 1]