Pyspark Select Array Element