Python Select Function In Pyspark Interview