如何在pyspark中将dataframe行的每个值除以行的总和(数据规范化)？

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

spark = SparkSession.builder.getOrCreate()

data = [(1, 2, 3), (4, 5, 6), (7, 8, 9)]
df = spark.createDataFrame(data, ["col1", "col2", "col3"])

df_with_sum = df.withColumn("row_sum", sum(col(col_name) for col_name in df.columns))

normalized_df = df_with_sum.select(
    *[col(col_name) / col("row_sum") for col_name in df.columns]
)