可以使用mappartitions：

def func(iterator):
    print("hello world")
    return iterator

stitching_result = sc.\
    parallelize(a, partitions).\
    mapPartitions(func, True).\
    map(lambda l: stitch_images_pair(project, l)).\
    collect()

打印五次 hello world 执行前 stitch_images_pair .
在内部使用迭代器时 func 迭代器应该具体化为一个列表，并返回一个新的迭代器。以下代码打印每个分区中的行数：

def func(iterator):
    data = list(iterator)
    print(len(data))
    return iter(data)