ruby の inject をわかりやすく説明してみる

ruby の inject って慣れないと少し理解しづらいよなーと思ったので、極力わかりやすい説明をしてみるテスト。

さて、1 から 10 までの合計を求めるこんな↓コードがあった場合

sum = 0
(1..10).each {|i| sum = sum + i }
p sum # => 55

inject を使ってこのよう↓に書けます。

p (1..10).inject(0) {|sum, i| sum + i }

each と inject でどのように書き変わってるかを図で示すとこんな↓感じ。

injectの引数 0 は、ブロックローカルな sum 変数の初期値になってます。で、ブロックの実行結果の値が sum に代入されて、２回目以降のループを実行します。ループしている間の、各変数とブロックの中身はこんな↓感じ。

最後に実行されたブロックの結果が、inject の戻り値となります。

んじゃ、inject で書き換える事ができる他のパターンを見てみます

こんな↓コードがあった場合

fib = [1, 1]
(0..10).each {|i| fib << fib[i] + fib[i+1]}
p fib # => [1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233]

inject を使ってこのよう↓に書けます。

p (0..10).inject([1, 1]) {|fib, i| fib << fib[i] + fib[i+1] }

こんな↓コードがあった場合

data = [:A, :B, :A, :C, :E, :A, :D, :B, :B, :C, :E]
h = Hash.new(0)
data.each {|key| h[key] += 1 }
p h # => {:A=>3, :B=>3, :C=>2, :E=>2, :D=>1}

inject を使ってこのよう↓に書けます。

data = [:A, :B, :A, :C, :E, :A, :D, :B, :B, :C, :E]
p data.inject(Hash.new(0)) {|h, key| h[key] += 1; h }

上の例の場合、ブロックの実行結果でハッシュ(h)を返したいので、セミコロンで区切って h を返しています。

どの例でも、ループで計算した結果を保持しとく変数が、inject では、ブロックローカルな変数で済んでるのでスッキリした感じがしますね。そんなわけで、皆で inject厨になりましょう。

Λάδι Βιώσας