views:

121

answers:

1

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?

+4  A: 
def inventory
  Hash[pieces.group_by(&:flavor).map{|f,p| [f, p.size]}]
end
mckeed
Nice. Never even knew group_by existed.
bergyman
It's new to Ruby 1.8.7. Just `require "backports"` if using 1.8.6. BTW, `Hash[key_value_pairs]` is also new to 1.8.7
Marc-André Lafortune
Using Ruby 1.9, thanks :)
macek