Pyspark First Element Of Array, Finally, use collect_list to create an array of the first elements.