替换Pyspark中的多个元素

from pyspark.sql.functions import when, col

df = df.withColumn('column_name', when(col('column_name') == 'value1', 'replacement1')
                                  .when(col('column_name') == 'value2', 'replacement2')
                                  .otherwise(col('column_name')))

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

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

# 创建示例数据
data = [('Alice', 25), ('Bob', 30), ('Charlie', 35)]
df = spark.createDataFrame(data, ['name', 'age'])

# 替换多个元素
df = df.withColumn('name', when(col('name') == 'Alice', 'A')
                            .when(col('name') == 'Bob', 'B')
                            .when(col('name') == 'Charlie', 'C')
                            .otherwise(col('name')))

# 显示结果
df.show()