如何为数组中所有先前的值添加值

Question

假设我有以下数组：

my_array = [1, 5, 8, 11, -6]

我需要迭代该数组并将当前值之前的值添加在一起。举个例子可能会更容易理解。我需要返回一个看起来像这样的数组：

final_array = [1, 6, 14, 25, 19]

我尝试过这样做：

my_array.collect {|value| value + previous_values }

但显然这不起作用，因为我不知道如何获取数组中以前的值。

我是一个编程新手，所以这可能比我做的更容易。我很确定我需要使用收集或注入，但我似乎不知道如何做到这一点。

任何帮助将不胜感激。

Answer 1

我的第一直觉是：“这显然是一次扫描（又名前缀求和），所以这应该很容易”：

[1, 5, 8, 11, -6].scan(:+)

显然，我最近读了太多的 Haskell 和 Scala，因为 没有 Ruby 中的 Enumerable#scan ...但是：

module Enumerable
  def scan(initial=first, &block)
    [initial].tap {|res| 
      reduce {|acc, el| 
        block.(acc, el).tap {|el|
          res << el
        }
      }
    }
  end
end

如果您希望 Enumerable#scan 表现得像 Enumerable#reduce，即采用可选的初始参数并一个可选符号，我们需要使用从 Rubinius 的 Enumerable#reduce 窃取的一些参数按摩代码来稍微增强我们的版本：

module Enumerable
  def scan(initial=nil, sym=nil, &block)
    args = if initial then [initial] else [] end
    unless block_given?
      args, sym, initial = [], initial, first unless sym
      block = ->(acc, el) { acc.send(sym, el) }
    end
    [initial || first].tap {|res| 
      reduce(*args) {|acc, el| 
        block.(acc, el).tap {|e|
          res << e
        }
      }
    }
  end
end

有了这个增强版本，上面的示例现在可以工作了：

p [1, 5, 8, 11, -6].scan(:+)
# => [1, 6, 14, 25, 19]

如果您再次遇到此类问题，在另一种语言中，请记住术语 scan 和 prefix-sum，此类函数通常非常常见。我不太明白为什么 Ruby 还没有它们。

Answer 2

您自己使用 collect 进行的尝试已经非常接近了；只要继续对之前的值进行求和就可以了。

my_array = [1, 5, 8, 11, -6]
previous_values = 0
my_array.collect { |value| previous_values += value }
# => [1, 6, 14, 25, 19]

Answer 3

x = 0
[1, 5, 8, 11, -6].map {|a| x = x +a }

Answer 4

你可以使用这个：

my_array = [1, 5, 8, 11, -6]
final_array = []

my_array.inject(0) {|res, it| final_array << res + it; res + it}

Answer 5

my_array.each_index{|i| my_array[i] += my_array[i-1] if i>0 }

或者

my_array.inject([]){|memo, item| memo << item + memo.last.to_i }

Answer 6

我为此制作了一个预分配结果数组的 gem。即使对于具有大量元素的 Enumerables，操作也非常快。与使用 Enumerable#map 的解决方案不同，其语法与 Enumerable#reduce 的语法完全相同，并且可以选择在幕后使用 Enumerable#reduce，以防您使用猴子修补的 #reduce。该名称取自 Clojure 的同名函数。

https://rubygems.org/gems/reductions

安装：

$ gem install reductions

使用：

require 'reductions'

[1, 5, 8, 11, -6].reductions(:+)            #=> [1, 6, 14, 25, 19]
[1, 5, 8, 11, -6].reductions{|a| a + b}     #=> [1, 6, 14, 25, 19]

# With an inital value:
[1, 5, 8, 11, -6].reductions(1,:+)          #=> [1, 2, 7, 15, 26, 20]
[1, 5, 8, 11, -6].reductions(1){|a| a + b}  #=> [1, 2, 7, 15, 26, 20]