How to use map and reduce in pyspark for list object with some list variables