Pyspark Reduce By Example