Pyspark Split Array Javascript