ruby: suma odpowiadających członków dwóch lub więcej tablic

Question

Mam dwie (lub więcej) tablice z 12 liczbami całkowitymi w każdej (odpowiada wartości dla każdego miesiąca). Wszystko czego chcę, to dodać je do siebie, dzięki czemu mam jedną tablicę z sumowanymi wartościami dla każdego miesiąca. Poniżej znajduje się przykład z trzech wartości: [1,2,3] i [4,5,6] => [5,7,9]

Najlepszym może wymyślić był następujący:

[[1,2,3],[4,5,6]].transpose.map{|arr| arr.inject{|sum, element| sum+element}} #=> [5,7,9] 

Czy jest lepszy sposób to zrobić? Po prostu wydaje się, że tak podstawową rzeczą jest chęć zrobienia.

Answer 1

Oto transpose wersja Anurag zasugerował:

[[1,2,3], [4,5,6]].transpose.map {|x| x.reduce(:+)} 

ta będzie współpracować z dowolną liczbą tablic składowych. reduce i inject są synonimami, ale reduce wydaje mi się bardziej wyraźnie komunikować intencje Kodeksu tu ...

Answer 2

oto moja próba code-golfa to rzecz:

// ruby 1.9 syntax, too bad they didn't add a sum() function afaik 
[1,2,3].zip([4,5,6]).map {|a| a.inject(:+)} # [5,7,9] 

zip powraca [1,4], [2,5], [3,6] i map sumy każdy sub-array.

Answer 3

[[1,2,3],[4,5,6]].transpose.map{|a| a.sum} #=> [5,7,9] 

Answer 4

Dla jaśniejszego składni (nie najszybszy), można skorzystać z Vector:

require 'matrix' 
Vector[1,2,3] + Vector[4,5,6] 
=> Vector[5, 7, 9] 

dla wielu wektorów, można zrobić:

arr = [ Vector[1,2,3], Vector[4,5,6], Vector[7,8,9] ] 
arr.inject(&:+) 
=> Vector[12, 15, 18] 

Jeśli chcesz załadować tablice do wektorów i suma:

arrays = [ [1,2,3], [4,5,6], [7,8,9] ] 
arrays.map { |a| Vector[*a] }.inject(:+) 
=> Vector[12, 15, 18] 

Answer 5

@FriendFX, masz rację pod względem odpowiedzi @ user2061694. W moim przypadku działało to tylko w środowisku Rails. można je uruchomić w czystym Ruby jeśli dokonać następujących zmian ...

W IRB

[[0, 0, 0], [2, 2, 1], [1,3,4]].transpose.map {|a| a.inject(:+)} 
=> [3, 5, 5] 


[[1,2,3],[4,5,6]].transpose.map {|a| a.inject(:+)} 
=> [5, 7, 9] 

Answer 6

To może nie być najlepszą odpowiedź, ale to działa.

array_one = [1,2,3] 
array_two = [4,5,6] 
x = 0 
array_three = [] 
while x < array_one.length 
    array_three[x] = array_one[x] + array_two[x] 
    x += 1 
end 

=>[5,7,9] 

To może być więcej linii kodu niż inne odpowiedzi, ale jest to odpowiedź mimo

Answer 7

Pokornie czuć, że pozostałe odpowiedzi widzę są tak skomplikowane, że będą one mylące dla recenzentów kodu. Trzeba dodać komentarz objaśniający, który po prostu zwiększa ilość potrzebnego tekstu.

Jak walka to zamiast:

a_arr = [1,2,3] 
b_arr = [4,5,6] 
(0..2).map{ |i| a_arr[i] + b_arr[i] } 

Nieco inne rozwiązanie: (tak, że nie jesteś ciężko kodowania "2")

a_arr = [1,2,3] 
b_arr = [4,5,6] 
c_arr = [] 
a_arr.each_index { |i| c_arr[i] = a_arr[i] + b_arr[i] } 

Wreszcie matematycznie rzecz biorąc, jest to ta sama pytanie jak to:

How do I perform vector addition in Ruby?

Answer 8

Dla:

a = [1,2,3] 
b = [4,5,6] 

Mogłeś zip a następnie użyć reduce:

p a.zip(b).map{|v| v.reduce(:+) } 
#=> [5, 7, 9] 

Lub, jeśli jesteś pewien, że tablica a i b zawsze będzie jednakowej długości:

p a.map.with_index { |v, i| v + b[i] } 
#=> [5, 7, 9] 

Answer 9

Teraz może używać sum w 2.4

nums = [[1, 2, 3], [4, 5, 6]] 
nums.transpose.map(&:sum) #=> [5, 7, 9]