如何在2个有序列的组上添加行索引，以便为每个组重新启动行索引，对于pyspark数据帧？

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

spark = SparkSession.builder.getOrCreate()

data = [("group1", 1), ("group1", 2), ("group1", 3), ("group2", 4), ("group2", 5), ("group2", 6)]
df = spark.createDataFrame(data, ["group", "value"])
df.show()

+------+-----+
| group|value|
+------+-----+
|group1|    1|
|group1|    2|
|group1|    3|
|group2|    4|
|group2|    5|
|group2|    6|
+------+-----+

windowSpec = Window.partitionBy("group").orderBy("value")
df_with_index = df.withColumn("index", row_number().over(windowSpec))
df_with_index.show()

+------+-----+-----+
| group|value|index|
+------+-----+-----+
|group1|    1|    1|
|group1|    2|    2|
|group1|    3|    3|
|group2|    4|    1|
|group2|    5|    2|
|group2|    6|    3|
+------+-----+-----+