Comment by __mharrison__

I looked at your code very quickly, but it looks like you need to use .filter after a .groupby...