将文件名输出到pyspark中的最终文件夹

from pyspark.sql import SparkSession
import os

spark = SparkSession.builder.getOrCreate()

file_path = "输入文件路径"
file_names = [os.path.basename(file) for file in os.listdir(file_path)]

output_path = "输出文件夹路径"
output_file = os.path.join(output_path, "output.txt")

with open(output_file, "w") as f:
    for file_name in file_names:
        f.write(file_name + "\n")