Ruby- Ajout/soustraction éléments d'un tableau avec un autre tableau

Question

Je fais ceci:

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

Je reçois cette réponse ->[1]
Je veux que cette réponse ->[1,1,1,1] (comme l'addition de la matrice/soustraction)

J'ai essayé ceci:

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

mais je reçois cette réponse: [1,0,0,0]

Quelqu'un peut-il me donner une bonne façon de faire cela? Merci beaucoup!

Answer 1

Vous pouvez utiliser zip:

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

Il y a aussi une classe 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

Vous pouvez utiliser each_with_index et map.

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