`zipWithIndex map` to `zipWithIndex.map`. Please check
https://cwiki.apache.org/confluence/display/SPARK/Spark+Code+Style+Guide
for infix usage.