如何在不使用databricks lib的情况下向Spark SQL查询结果添加列标题？

from pyspark.sql import SparkSession

# 创建SparkSession
spark = SparkSession.builder.getOrCreate()

# 执行Spark SQL查询并获取结果数据集
query_result = spark.sql("SELECT * FROM table_name")

# 获取查询结果数据集的schema
schema = query_result.schema

# 将查询结果数据集转换为RDD
query_result_rdd = query_result.rdd

# 创建一个包含列标题的RDD
header_rdd = spark.sparkContext.parallelize([schema.names])

# 将查询结果RDD与列标题RDD进行合并
merged_rdd = header_rdd.union(query_result_rdd)

# 将合并后的RDD转换回DataFrame
result_with_header = spark.createDataFrame(merged_rdd, schema)

# 使用带有列标题的DataFrame进行进一步的数据处理或分析
result_with_header.show()