迭代Pyspark dataframe的列，并根据条件填充新列

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

spark = SparkSession.builder.appName("Column Iteration").getOrCreate()

data = [("John", 25), ("Alice", 30), ("Bob", 35)]
df = spark.createDataFrame(data, ["Name", "Age"])
df.show()

+-----+---+
| Name|Age|
+-----+---+
| John| 25|
|Alice| 30|
|  Bob| 35|
+-----+---+

new_df = df
columns = df.columns

for column in columns:
    new_column = column + "_new"
    new_df = new_df.withColumn(new_column, when(df[column] > 30, "High").otherwise("Low"))

new_df.show()

+-----+---+--------+--------+
| Name|Age|Name_new|Age_new |
+-----+---+--------+--------+
| John| 25|     Low|     Low|
|Alice| 30|     Low|     Low|
|  Bob| 35|    High|    High|
+-----+---+--------+--------+