Pyspark:使用窗口函数按日期将数据帧保存到单个csv？

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, date_format, window

# 创建SparkSession
spark = SparkSession.builder.appName("WindowFunctionExample").getOrCreate()

# 读取数据集
df = spark.read.csv("data.csv", header=True, inferSchema=True)

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

# 定义窗口规范
windowSpec = window.orderBy(col("date")).rangeBetween(-1, 0)

# 使用窗口函数按日期分组并保存到单个CSV文件
df.withColumn("window", windowSpec).groupBy("window").csv("output.csv")

# 关闭SparkSession
spark.stop()