PySpark: implementing subtractByKey(), subtract() and keyBy()