Pyspark Reduce By Example Or For Example