groupby.apply is often the thing that is already way too slow though based on use-case - because I think it pretty much does what you wrote rather than somethin

groupby.apply is often the thing that is already way too slow though based on use-case - because I think it pretty much does what you wrote rather than something that makes stronger optimisation assumptions.