Applying Functions per Subgroups with Pandas: A Comprehensive Solution