Spark将数据加载到分区中，每个分区都在一个工作节点中。Spark提供了一个名为foreachPartition的分区迭代函数，你可以用它来打印每个分区的内容和与之关联的工作节点，下面是一个伪代码：

from pyspark.sql.functions import col
from pyspark import TaskContext

def print_partition_data(data):
    node_name = TaskContext.get().getHost()
    partition_id = TaskContext.getPartitionId()
    print(f"Partition number {partition_id} is on node {node_name}:")
    for d in data:
        print(d)

# Then call the function like this
df.foreachPartition(print_partition_data)