spark foreachPartition，如何获取每个分区的索引？

# 对RDD使用mapPartitionsWithIndex方法
def process_partition(index, iterator):
    # 在这里可以使用分区索引进行特定的操作
    for item in iterator:
        # 处理每个分区中的元素
        print("Partition Index:", index)
        print("Element:", item)

rdd = sc.parallelize([1, 2, 3, 4, 5], 3)
rdd.mapPartitionsWithIndex(process_partition).collect()

# 对DataFrame使用mapPartitionsWithIndex方法
def process_partition(index, iterator):
    # 在这里可以使用分区索引进行特定的操作
    for row in iterator:
        # 处理每个分区中的行
        print("Partition Index:", index)
        print("Row:", row)

df = spark.createDataFrame([(1, "A"), (2, "B"), (3, "C"), (4, "D"), (5, "E")], ["id", "value"])
df.rdd.mapPartitionsWithIndex(process_partition).collect()