Grouped output on Pyspark

Question

I\'m running the following code:

output = (df
          .groupby(\'customer_id\')
          .agg(
            f.countDistinct(\'customer_id\').alias(\'identif