Pyspark Select First Element From Array