pyspark.TaskContext.partitionId

TaskContext.partitionId() → int

The ID of the RDD partition that is computed by this task.

pyspark.TaskContext.getLocalProperty pyspark.TaskContext.resources