在其他列上按条件封顶一个pyspark列？

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

spark = SparkSession.builder.getOrCreate()

data = [(1, 10, 20), (2, 15, 25), (3, 30, 40)]
df = spark.createDataFrame(data, ["id", "col1", "col2"])

max_value = 25  # 设置封顶值

df = df.withColumn("col2", when(col("col2") > max_value, max_value).otherwise(col("col2")))