Pyspark:如何在日期上创建窗口

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

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

# 创建示例数据集
data = [("2022-01-01", 10),
        ("2022-01-02", 20),
        ("2022-01-03", 30),
        ("2022-01-04", 40),
        ("2022-01-05", 50)]

df = spark.createDataFrame(data, ["date", "value"])

# 将日期列转换为日期类型
df = df.withColumn("date", col("date").cast("date"))

# 创建窗口
window = Window.orderBy("date")

# 在窗口上使用日期函数进行操作
df = df.withColumn("row_number", row_number().over(window))

# 打印结果
df.show()

+----------+-----+----------+
|      date|value|row_number|
+----------+-----+----------+
|2022-01-01|   10|         1|
|2022-01-02|   20|         2|
|2022-01-03|   30|         3|
|2022-01-04|   40|         4|
|2022-01-05|   50|         5|
+----------+-----+----------+