Thrust Reduce By Key In Pyspark