How to get the number of elements in partition

Question

I'm exploring apache spark and wanted to know if there's any way to get the number of elements in a particular RDD partition using the partition ID?

Help required as it will make my tasks very easy.

Thanks in Advance

Answer 1

rdd.mapPartitions(iter => Array(iter.size).iterator, true) 

This command will give you a new RDD with elements that are the sizes of each partition