如何在一个spark dataframe的多个列上旋转？

from pyspark.sql import SparkSession
from pyspark.sql.functions import expr

spark = SparkSession.builder.appName("ColumnPivot").getOrCreate()

data = spark.createDataFrame([(1, "A", 10), (2, "B", 20), (3, "A", 30), (4, "B", 40), (5, "A", 50)], ["id", "category", "value"])
data.show()

+---+--------+-----+
| id|category|value|
+---+--------+-----+
|  1|       A|   10|
|  2|       B|   20|
|  3|       A|   30|
|  4|       B|   40|
|  5|       A|   50|
+---+--------+-----+

pivoted_data = data.groupBy("id").pivot("category").agg(expr("coalesce(first(value), 0)"))
pivoted_data.show()

+---+----+----+
| id|   A|   B|
+---+----+----+
|  1|  10|   0|
|  3|  30|   0|
|  5|  50|   0|
|  2|   0|  20|
|  4|   0|  40|
+---+----+----+