Narrow Transformations In Pyspark