https://bagelmobestcheapprice.blogspot.com/2024/06/apply-custom-function-to-groupby-in-vaex.html