Array Type In Pyspark