Spark Sql Array Functions Python