https://onestopde.com/pyspark-dataframe-transformations/