如何计算每个键在PySpark数据帧中的百分位数？

from pyspark.sql import SparkSession
from pyspark.sql.functions import expr

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

# 创建示例数据帧
data = [("A", 1), ("A", 2), ("A", 3), ("B", 4), ("B", 5), ("B", 6)]
df = spark.createDataFrame(data, ["key", "value"])

# 计算每个键的百分位数
percentiles = [25, 50, 75]  # 设置要计算的百分位数
exprs = [expr(f"percentile_approx(value, {p}/100)") for p in percentiles]  # 构建表达式
result = df.groupBy("key").agg(*exprs)  # 按键分组并计算百分位数

# 显示结果
result.show()