在Apache Spark中，可以使用mapPartitionsWithIndex函数来获取每个分区的第一行和最后一行。下面是一个示例代码：

from pyspark import SparkContext

def get_first_last_lines(iterator):
    first_line = next(iterator)
    last_line = None
    for line in iterator:
        last_line = line
    return [first_line, last_line]

sc = SparkContext("local", "FirstLastLines")
data = sc.textFile("input.txt")

first_last_lines = data.mapPartitionsWithIndex(get_first_last_lines).collect()
for partition, lines in enumerate(first_last_lines):
    print("Partition {}: First Line: {}, Last Line: {}".format(partition, lines[0], lines[1]))

在上面的示例代码中，我们首先创建了一个SparkContext对象，并将输入数据加载为RDD。然后，我们定义了一个名为get_first_last_lines的函数，该函数接收一个分区的迭代器作为参数，并返回该分区的第一行和最后一行。

接下来，我们使用mapPartitionsWithIndex函数将get_first_last_lines函数应用于每个分区，并使用collect函数将结果收集回驱动程序。最后，我们循环遍历每个分区的第一行和最后一行，并打印出来。

请注意，此代码示例假设输入数据是文本文件，并且每行包含一个记录。如果你的输入数据不是这种格式，你需要根据实际情况修改代码来适应你的数据格式。