在pyspark中将不同事件的计数结果拆分为不同的列

from pyspark.sql import SparkSession
from pyspark.sql.functions import count

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

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

# 使用groupBy和count函数计算每个事件的计数结果
count_df = data.groupBy("事件").agg(count("*").alias("计数"))

# 使用pivot函数将不同事件的计数结果拆分为不同的列
pivot_df = count_df.groupBy().pivot("事件").sum("计数")

# 显示结果
pivot_df.show()