https://tanvirislam3113.blogspot.com/2024/09/add-column-to-pyspark-dataframe-based.html