Pyspark Select First Element Of Array Column