Spark Sql Create Array In Python