pyspark.RDD.mapPartitionsWithIndex¶
-
RDD.
mapPartitionsWithIndex
(f: Callable[[int, Iterable[T]], Iterable[U]], preservesPartitioning: bool = False) → pyspark.rdd.RDD[U]¶ Return a new RDD by applying a function to each partition of this RDD, while tracking the index of the original partition.
Examples
>>> rdd = sc.parallelize([1, 2, 3, 4], 4) >>> def f(splitIndex, iterator): yield splitIndex >>> rdd.mapPartitionsWithIndex(f).sum() 6