Ruby 2.2 で追加されたメソッドまとめをみてた。
New Methods in Ruby 2.2
Enumerable#slice_whenのユースケースでEnumerable内での各要素出現数をカウントするのがステキだったので、2.2以前も含めてどう書けるパターンあるんだっけ？をメモ。
カウント対象のArrayを生成
labels = ["aaa", "bbb", "ccc"]
ary = Array.new(20) { labels.sample }
puts ary.join ", "
# bbb, aaa, bbb, bbb, bbb, aaa, ccc, aaa, aaa, bbb, ccc, ccc, bbb, bbb, aaa, aaa, ccc, bbb, aaa, aaa
2.2以前
Enumerable#injectを利用
r1 = ary.inject(Hash.new(0)) {|h, e| h.tap {|h| h[e] += 1} }
puts r1
# {"bbb"=>8, "aaa"=>8, "ccc"=>4}
2.2から
Enumerable#slice_whenを利用
あえて出力結果を Hash に寄せてみる。
r2 = ary.sort.slice_when(&:!=).map {|x| [x.first, x.size] }
puts Hash[r2]
# {"aaa"=>8, "bbb"=>8, "ccc"=>4}
Enumerable#group_by + Object#itself でもイケるな
r3 = ary.group_by(&:itself).map {|k, v| [k, v.size] }
puts Hash[r3]
# {"bbb"=>8, "aaa"=>8, "ccc"=>4}
メモ
-
Enumerable#injectは副作用ベースな書き方なのでいつも一瞬「うっ」ってなってる -
Enumerable#slice_whenは#sortしないといけないね - 要素数が大きくなった場合に計算量とメモリ使用量の観点で効率的なのはどれだろう？
- 富豪的な実装でもいいから簡単に書きたい時もあるよね、使い分けたい
コード全体
labels = ["aaa", "bbb", "ccc"]
ary = Array.new(20) { labels.sample }
puts ary.join ", "
r1 = ary.inject(Hash.new(0)) {|h, e| h.tap {|h| h[e] += 1} }
puts r1
r2 = ary.sort.slice_when(&:!=).map {|x| [x.first, x.size] }
puts Hash[r2]
r3 = ary.group_by(&:itself).map {|k, v| [k, v.size] }
puts Hash[r3]