Add distinct count of a column to each row in PySpark

Question

I need to add distinct count of a column to each row in PySpark dataframe.

Example: If the original dataframe is this:

+----+----+
|col1|col2|
+----+--