Rubyで配列の数字を足していきたい

Question

Arrayに累積配列？を作成したいですがよい実装方法を教えてください

例
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]から
[1, 3, 6, 10, 15, 21, 28, 36, 45, 55]を返したいです。

Answer 1

効率重視

(1..10).inject([]){|a, v| a << (a.last || 0) + v; a}

シンプルさ重視

10.times.map{|i| (1..i+1).inject(:+)}

Answer 2

計算は以下のように行えます。

ns = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]  # 入力となる Array オブジェクト ns
f = -> n { (1..n).reduce :+ }         # 1 から n までの総和を求める手続き f
ns.map &f                             # ns の各要素に f を適用

上の３行は、以下と同じことです。

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10].map {|n| (1..n).reduce :+ }

「Array に〜実装」したいとのことですが、実装方針によって以下のような場合があると思います。

Array クラスのインスタンスメソッドとして実装する場合

class Array
  def calc
    map {|n| (1..n).reduce :+ }
  end
end

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10].calc  #=> [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]

オブジェクト ns の特異メソッドとして実装する場合

ns = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

def ns.calc
  map {|n| (1..n).reduce :+ }
end

ns.calc #=> [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]

入力となる Array オブジェクトを引数に指定する手続きとして実装する場合

calc = -> ns { ns.map {|n| (1..n).reduce :+ } }

calc.([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]) #=> [1, 3, 6, 10, 15, 21, 28, 36, 45, 55]

Answer 3

a=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b="0"
c=a.map{|x| b = b + "+" + x.to_s}.map{|x|eval(x)}
p c

[0+1,0+1+2,...,0+1+...+10]を作ってevalで計算してみたり。

Answer 4

CertaiNさんと同様の回答になりますが、以下でどうでしょうか？

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
p a.size.times.map{|i| a[0..i].inject(:+)}

Comments

[mpyw]

なるほど、あらかじめ `a` が存在してるならこれですね

[mpyw]

# コメント欄で ` ` 使えないのか…あと消せないのがつらい(汗

[manman]

「1からNまでの配列」に限らず、一般の配列としたかったので、
a とおきました。

a = [1, 2, 3.2, 4, 5]
p a.size.times.map{|i| a[0..i].inject(:+)}
#=> [1, 3, 6.2, 10.2, 15.2]

余談ですが、コメント欄はMarkdownが使えないみたいです。

[otn]

CertaiNさんの「あらかじめ a が存在しているなら」というのは、
メソッドチェーンの中で使うなら、下記のようにいったん変数に置かないといけないというような意味でしょう。
(a=foo.bar.baz).size.times.map{|i| a[0..i].inject(:+)}.hoge.hage

> 「1からNまでの配列」に限らず、一般の配列としたかったので、
当然そういう質問だと思っていたのですが。

[tamy]

トリッキーですが tap 使えば一時変数消せますね

[1,2,3,4,5].tap{|a| break a.size.times.map{|i| a[0..i].inject(:+)}}

Answer 5

一番シンプル。

a=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
x=0
b=a.map{|w| x+=w}

CertaiNさんの1つめの改造。ループ内で先頭の場合の特別処理を無くしました。

a=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b=a.inject([0]){|x,v| x<<x.last+v}
b.shift

a=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
b=a.inject([0]){|x,v| x<<x.last+v}.drop(1)

(変数名が被って直感的にわかりにくいのでブロック引き数名を変更。今のRubyだと被ってもいいのですが）
修正：最後の例、dropというメソッドがあったので、[1..-1]を.drop(1)に書き換えた。

Answer 6

問題を勘違いしていました。
回答自体を削除しようとおもいましたが、方法がわからず...

-------- 以下は無視してください ... -----
配列の合計を得るというのは、よくある処理ですね。たくさんの情報を google 検索でえることができます。
いくつかを紹介します。

• Rubyで配列に入っている数値の合計を出すメソッドを作る(spec付き) http://qiita.com/kakkunpakkun/items/eac4e8b76978b4a65c26

• Rubyで数値の配列の合計を求める方法 http://www.xmisao.com/2014/03/25/how-to-sum-array-of-numbers-in-ruby.html

a = [123, 456, 789]
p a.inject(:+) #=> 1368

• sum(identity = 0, &block) : Calculates a sum from the elements http://apidock.com/rails/Enumerable/sum

Rails 中には sum() というメソッドがあります。

[5, 15, 10].sum # => 30

Answer 7

こちらでどうでしょうか？

class Array
def cumulative_sum
sum = 0
self.map{|x| sum += x}
end
end

a = (1..10).to_a
a.cumulavive_sum