Stack Function In Pyspark