在PySpark中计算Spark DataFrame多个列值的百分比

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

spark = SparkSession.builder.getOrCreate()

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

total_col1 = df.agg(sum(col("col1"))).collect()[0][0]
total_col2 = df.agg(sum(col("col2"))).collect()[0][0]

df_with_percentage = df.withColumn("col1_percentage", col("col1") / total_col1 * 100)\
                      .withColumn("col2_percentage", col("col2") / total_col2 * 100)

df_with_percentage.show()