I have a pyspark dataframe which contains the data similar to below:
id class price place 1 A 10 US 2 B 5 US 3 B 5 MEX