如何使用Spark Dataframe中的表达式合并重复行

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

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

data = [("John", 25, "USA"), ("Alice", 30, "Canada"), ("John", 35, "USA")]
df = spark.createDataFrame(data, ["Name", "Age", "Country"])
df.show()

merged_df = df.groupBy("Name").agg(expr("collect_list(Age) as Age"), expr("collect_list(Country) as Country"))
merged_df.show()