使用pyspark在同一组中查找多个最大日期

from pyspark.sql import SparkSession
from pyspark.sql.functions import col, max

spark = SparkSession.builder.appName("MaxDate").getOrCreate()

data = [("group1", "2022-01-01"),
        ("group1", "2022-02-01"),
        ("group2", "2022-03-01"),
        ("group2", "2022-04-01"),
        ("group2", "2022-05-01")]

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

max_dates = df.groupBy("group").agg(max(col("date")).alias("max_date"))

max_dates.show()