如何在对行执行映射操作后保留Spark Dataframe的列结构

from pyspark.sql import SparkSession
from pyspark.sql.functions import col

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

# 创建原始Dataframe
data = [("Alice", 25), ("Bob", 30), ("Charlie", 35)]
df = spark.createDataFrame(data, ["name", "age"])

# 定义对行映射的函数
def map_func(row):
    name = row["name"]
    age = row["age"]
    # 在这里执行对行的映射操作
    # ...

    # 返回一个新的行
    return (name, age)

# 使用select函数保留列结构
new_df = df.select(*[col(c) for c in df.columns], col("name").alias("new_name"), col("age").alias("new_age"))

# 显示新的Dataframe
new_df.show()