Spark Sql Array Size Python