myUdf = udf(myFunc, StringType())
df.withColumn('data', concat_ws(',', col('B'), col('C'))).groupBy('A').agg(
    collect_list('data').alias('data'))
Применение UDF к GroupedData в PySpark (с функционирующим примером Python)