Postgresql Count Distinct In Pyspark