Postgresql Array Agg Function In Pyspark