在pyspark中使用函数和字典重新转换列类型

from pyspark.sql import SparkSession
from pyspark.sql.functions import col
from pyspark.sql.types import StringType, IntegerType, DoubleType

spark = SparkSession.builder.getOrCreate()

df = spark.read.csv("path/to/dataset.csv", header=True, inferSchema=True)

type_mapping = {
    "column1": IntegerType(),
    "column2": StringType(),
    "column3": DoubleType()
}

for column, data_type in type_mapping.items():
    df = df.withColumn(column, col(column).cast(data_type))

df.show()