Pyspark Get Values Python