spark导出csv附带schema

你可以使用以下代码将DataFrame导出为带有模式的CSV文件：

from pyspark.sql.functions import *

# 创建一个DataFrame
df = spark.createDataFrame([(1, "John"), (2, "Jane"), (3, "Jim")], ["id", "name"])

# 将DataFrame导出为CSV文件
df.coalesce(1).write.format("csv").option("header", "true").mode("overwrite").csv("output.csv")

# 读取CSV文件并将其转换为DataFrame，同时保留模式
df_with_schema = spark.read.option("header", "true").csv("output.csv")
df_with_schema.show()

在此示例中，我们使用`createDataFrame`创建一个DataFrame，然后使用`coalesce(1)`将所有数据合并到单个分区中，最后使用`write`方法将数据导出为CSV文件。我们使用`option("header", "true")`指定输出文件应包含标题行，并使用`mode("overwrite")`覆盖任何现有的输出文件。然后，我们使用`read`方法读取CSV文件，并使用`option("header", "true")`指定输入文件包含标题行。Spark将自动推断模式并将其应用于读取的CSV文件中的数据。最后，我们使用`show`方法显示转换后的DataFrame以进行验证。