https://mobhitsbestcheapprice.blogspot.com/2024/01/custom-function-over-pyspark-dataframe.html