Postgres Count Distinct In Pyspark