Pyspark DF Pivot and Create Arrays columns