如何将groupBy和聚合函数应用于PySpark DataFrame中的特定窗口？

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

spark = SparkSession.builder.getOrCreate()

data = [(1, "A", 100), (1, "B", 200), (2, "A", 150), (2, "B", 250)]
df = spark.createDataFrame(data, ["id", "category", "value"])

windowSpec = Window.partitionBy("id").orderBy("category").rowsBetween(-1, 1)

result = df.withColumn("sum_value", sum(col("value")).over(windowSpec))

result.show()

+---+--------+-----+---------+
| id|category|value|sum_value|
+---+--------+-----+---------+
|  1|       A|  100|      300|
|  1|       B|  200|      500|
|  2|       A|  150|      450|
|  2|       B|  250|      400|
+---+--------+-----+---------+