如何在窗口内首次排序后保留Spark dataframe中的默认顺序

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

# 创建SparkSession
spark = SparkSession.builder.getOrCreate()

# 创建DataFrame，并添加保留默认顺序的列
df = spark.createDataFrame([(1, "A"), (2, "B"), (3, "C"), (4, "D")], ["id", "value"])

# 添加一个自增的列作为保留默认顺序的列
df = df.withColumn("default_order", row_number().over(Window.orderBy("id")))

# 定义窗口函数和排序函数
windowSpec = Window.partitionBy().orderBy("default_order")
sorted_df = df.withColumn("sorted_order", row_number().over(windowSpec))

# 输出结果
sorted_df.show()