ENH: speed up multi-key groupby