Select Columns Pyspark Dataframe Transformation