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