Python multicolumn groupby

CC @talumbau