Search code examples
rubyarrayshashenumerable

Calculate sum of objects for each unique object property in Ruby

I was helping with an answer in this question and it sparked a question of my own.

  1. Pie is an object that has a pieces array made of of PiePiece objects.
  2. Each PiePiece has a flavor attribute

How do I create a hash that looks like this:

# flavor => number of pieces
{
  :cherry => 3
  :apple => 1
  :strawberry => 2
}

This works, but I think it could be improved

def inventory
  hash = {}
  pieces.each do |p|
    hash[p.flavor] ||= 0
    hash[p.flavor] += 1
  end
  hash
end

Any ideas?

Solution 
  • def inventory
  Hash[pieces.group_by(&:flavor).map{|f,p| [f, p.size]}]
end