如何在pyspark中的一列上应用窗口函数？

from pyspark.sql import SparkSession
from pyspark.sql.window import Window
from pyspark.sql.functions import *

spark = SparkSession.builder.getOrCreate()

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

windowSpec = Window.partitionBy("category").orderBy("value")

df.withColumn("rank", rank().over(windowSpec)).show()

from pyspark.sql import SparkSession
from pyspark.sql.window import Window
from pyspark.sql.functions import *

spark = SparkSession.builder.getOrCreate()

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

windowSpec = Window.partitionBy("category").orderBy("value")

df.withColumn("rank", rank().over(windowSpec)).show()