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