Maybe we could add this as a histogram here https://github.com/Shopify/spark/blob/disable_combinebykey_metric/python/pyspark/rdd.py#L1000?