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]