Pyspark Reduce By Examples